megengine.functional.full_like

full_like(inp, value)[源代码]

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

参数
返回类型

Tensor

返回

输出张量。

实际案例

>>> import numpy as np
>>> inp = Tensor(np.arange(1, 7, dtype=np.int32).reshape(2,3))
>>> F.full_like(inp, 2)
Tensor([[2 2 2]
 [2 2 2]], dtype=int32, device=xpux:0)