megengine.functional.ones_like¶

使用输入 Tensor 来填充输出 Tensor，保持相同的形状和数据类型。

参数: inp (Tensor) – input tensor from which to derive the output tensor shape.
返回类型: Tensor
返回: a tensor having the same shape as input tensor and filled with ones.

实际案例

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

megengine.functional.ones

megengine.functional.full