mindspore.mint.rand_like

mindspore.mint.rand_like(input, *, dtype=None, device=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 .
device (str, optional) – The specified device of the output tensor. Only "Ascend" and "npu" are supported. If device = None, the device of input will be used. Default: None .

Returns

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

Raises

RuntimeError – If Input device is "CPU", and device is None .
RuntimeError – If device is "CPU" .
ValueError – If device is "GPU" .

Supported Platforms:: Ascend

Examples

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