megengine.functional.ones_like

ones_like(inp)[源代码]

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

参数

inp (Tensor) – 输入张量。

返回类型

Tensor

返回

一个全1的tensor。

实际案例

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