megengine.functional.full_like¶

full_like(inp, value)[源代码]¶

返回一个与输入张量形状相同的且值为给定值的零张量

参数

返回类型

Tensor

返回

与输入张量形状相同且每个元素都等于填充值的张量。

实际案例

>>> x = F.arange(6, dtype='int32').reshape(2, 3)
>>> F.full_like(x, 2)
Tensor([[2 2 2]
 [2 2 2]], dtype=int32, device=xpux:0)

megengine.functional.full

megengine.functional.complex