mindspore.numpy.ones_like

mindspore.numpy.ones_like(a, dtype=None, shape=None)[source]

Returns an array of ones with the same shape and type as a given array.

Note

Input array must have the same size across a dimension. If a is not a Tensor, dtype is float32 by default if not provided.

Parameters

a (Union[Tensor, list, tuple]) – The shape and data-type of a define these same attributes of the returned array.
dtype (mindspore.dtype, optional) – Overrides the data type of the result. Default: None.
shape (int, sequence of ints, optional) – Overrides the shape of the result. Default: None.

Returns

Tensor, array of ones with the same shape and type as a.

Raises

ValueError – If a is not a Tensor, list or tuple.

Examples

>>> import mindspore.numpy as np
>>> a = np.ones((4,1,2))
>>> output = np.ones_like(a)
>>> print(output)
[[[1. 1.]]
[[1. 1.]]
[[1. 1.]]
[[1. 1.]]]