megengine.functional.full_like

full_like(inp, value)

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

参数:

• inp (Tensor) – 输入张量，从中导出输出张量形状。

• value (Scalar) – 填充值

返回类型:

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)