megengine.functional.ones_like

ones_like(inp)

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

参数

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

返回类型

Tensor

返回

与输入张量具有相同形状并填充为 1 的张量。

实际案例

>>> 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