mindspore.ops.ones

查看源文件
mindspore.ops.ones(shape, dtype=None)[源代码]

创建一个值全为1的tensor。

警告

参数 shape 在后续版本中将不再支持Tensor类型的输入。

参数:
  • shape (Union[tuple[int], list[int], int, Tensor]) - 指定的shape。

  • dtype (mindspore.dtype) - 指定的数据类型,默认 None

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> mindspore.ops.ones(4)
Tensor(shape=[4], dtype=Float32, value= [ 1.00000000e+00,  1.00000000e+00,  1.00000000e+00,  1.00000000e+00])
>>> mindspore.ops.ones((2, 3))
Tensor(shape=[2, 3], dtype=Float32, value=
[[ 1.00000000e+00,  1.00000000e+00,  1.00000000e+00],
 [ 1.00000000e+00,  1.00000000e+00,  1.00000000e+00]])
>>> mindspore.ops.ones(mindspore.tensor([1, 2, 3]))
Tensor(shape=[1, 2, 3], dtype=Float32, value=
[[[ 1.00000000e+00,  1.00000000e+00,  1.00000000e+00],
  [ 1.00000000e+00,  1.00000000e+00,  1.00000000e+00]]])