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)