mindspore.Tensor.new_ones

Tensor.new_ones(size, dtype=None) → Tensor[source]

Return a tensor of size filled with ones.

Warning

Non-backward-compatible change after version 2.9.0: parameter dtype will become keyword-only.

Parameters

  • size (Union[int, tuple(int), list(int)]) – An int, list or tuple of integers defining the output shape.

  • dtype (mindspore.dtype, optional) – The desired dtype of the output tensor. If None, the returned tensor has the same dtype as self. Default: None.

Returns

Tensor, the shape and dtype are defined above and filled with ones.

Raises

  • TypeError – If size is neither an int nor a tuple/list of int.

  • TypeError – If dtype is not a MindSpore dtype.

  • ValueError – If size contains negative values.

Supported Platforms:

Ascend GPU CPU

Examples

>>> import mindspore
>>> from mindspore import Tensor
>>> x = Tensor((), mindspore.int32)
>>> x.new_ones((2, 3))
Tensor(shape=[2, 3], dtype=Int32, value=
[[1, 1, 1],
 [1, 1, 1]])