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)