mindspore.mint.rand_like

View Source On Gitee
mindspore.mint.rand_like(input, *, dtype=None)[source]

Returns a new tensor that fills numbers from the uniform distribution over an interval \([0, 1)\) based on the given dtype and shape of the input tensor.

Parameters

input (Tensor) – Input Tensor to specify the output shape and its default dtype.

Keyword Arguments

dtype (mindspore.dtype, optional) – Designated tensor dtype, it must be float type. If None, the same dtype of input will be applied. Default: None .

Returns

Tensor, with the designated shape and dtype, filled with random numbers from the uniform distribution on the interval \([0, 1)\).

Raises

ValueError – If dtype is not a mstype.float_type type.

Supported Platforms:

Ascend

Examples

>>> import mindspore as ms
>>> from mindspore import Tensor, mint
>>> a = Tensor([[2, 3, 4], [1, 2, 3]])
>>> print(mint.rand_like(a, dtype=ms.float32).shape)
(2, 3)