mindspore.ops.ones_like

mindspore.ops.ones_like(input, *, dtype=None)

返回一个shape与输入相同，值为1的tensor。

参数：

• input (Tensor) - 输入tensor。

关键字参数：

• dtype (mindspore.dtype, 可选) - 指定数据类型。默认 None 表示与 input 数据类型相同。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input = mindspore.ops.arange(4)
>>> mindspore.ops.ones_like(input)
Tensor(shape=[4], dtype=Int64, value= [1, 1, 1, 1])
>>> mindspore.ops.ones_like(x, dtype=mindspore.float32)
Tensor(shape=[3, 2], dtype=Float32, value=
[[ 1.00000000e+00,  1.00000000e+00],
 [ 1.00000000e+00,  1.00000000e+00],
 [ 1.00000000e+00,  1.00000000e+00]])