mindspore.numpy.zeros_like
- mindspore.numpy.zeros_like(a, dtype=None, shape=None)[源代码]
返回一个与给定数组 a 具有相同shape和类型的Tensor,其中所有元素用0来填充。
说明
输入数组在同一维度上必须具有相同的size。如果 a 不是Tensor,且未设置 dtype ,那么默认情况下 dtype 是float32。
- 参数:
a (Union[Tensor, list, tuple]) - 原数组,返回数组的shape和数据类型与 a 相同。
dtype (mindspore.dtype, 可选) - 覆盖结果的数据类型。
shape (int, ints的序列, 可选) - 覆盖结果的shape。
- 返回:
Tensor,与给定 a 的shape和数据类型相同,其中所有元素都为0。
- 异常:
ValueError - 如果 a 不是Tensor、List或Tuple。
- 支持平台:
AscendGPUCPU
样例:
>>> import mindspore.numpy as np >>> a = np.ones((4,1,2)) >>> output = np.zeros_like(a) >>> print(output) [[[0. 0.]] [[0. 0.]] [[0. 0.]] [[0. 0.]]]