mindspore.ops.Ones

class mindspore.ops.Ones[源代码]

创建一个值全为1的Tensor。

第一个参数指定Tensor的shape,第二个参数指定填充值的数据类型。

输入:

  • shape (Union[tuple[int], int]) - 指定输出Tensor的shape,只能是正整数常量。

  • type (mindspore.dtype) - 指定输出Tensor的数据类型,只能是常量值。

输出:

Tensor,shape和数据类型与输入相同。

异常:

  • TypeError - shape 既不是tuple,也不是int。

支持平台:

Ascend GPU CPU

样例:

>>> ones = ops.Ones()
>>> output = ones((2, 2), mindspore.float32)
>>> print(output)
[[1. 1.]
 [1. 1.]]
>>> output = ones((3, 3), mindspore.float32)
>>> print(output)
[[1. 1. 1.]
 [1. 1. 1.]
 [1. 1. 1.]]