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.]]]