mindspore.numpy.ones

查看源文件
mindspore.numpy.ones(shape, dtype=mstype.float32)[源代码]

返回一个给定shape和类型的新Tensor,其中所有元素用1来填充。

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

  • dtype (Union[mindspore.dtype, str], 可选) - 指定的Tensor dtype 。默认值: mstype.float32

返回:

Tensor,给定 shapedtype ,其中所有元素都为1。

异常:
  • TypeError - 如果输入参数非上述给定的类型。

  • ValueError - 如果 shape 的元素项值 \(<0\)

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore.numpy as np
>>> print(np.ones((2,2)))
[[1. 1.]
[1. 1.]]