mindspore.ops.ones

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

创建一个值全为1的tensor。

警告

2.9.0(不含)之后版本非兼容性变更:shape 将改为 size,且 dtype 将改为关键字参数。当前版本中, 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]]])