mindspore.numpy.empty_like
- mindspore.numpy.empty_like(prototype, dtype=None, shape=None)[源代码]
返回一个shape和类型与给定数组相同的新数组。
说明
输入数组在整个维度上必须具有相同的大小。如果
prototype
不是Tensor,且未设置dtype
,则dtype
默认为float32。- 参数:
prototype (Union[Tensor, list, tuple]) - 原数组,其中生成数组的shape、类型默认和原数组相同。
dtype (mindspore.dtype, 可选) - 覆盖结果的数据类型。
shape (int, ints的序列, 可选) - 覆盖结果的shape。
- 返回:
Tensor,给定shape和类型与
prototype
相同的未初始化(任意)数据的数组。- 异常:
ValueError - 如果
prototype
不是Tensor、List或Tuple。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore.numpy as np >>> a = np.ones((4,1,2)) >>> output = np.empty_like(a) >>> print(output) [[[0. 0.]] [[0. 0.]] [[0. 0.]] [[0. 0.]]]