megengine.functional.zeros_like

zeros_like(inp)

返回一个跟输入tensor拥有相同shape和dtype的全零tensor。

参数:

inp (Tensor) – 输入张量，从中导出输出张量形状。

返回类型:

Tensor

返回:

一个和输入tensor拥有相同shape的全零tensor。

实际案例

>>> x = F.arange(6, dtype='int32').reshape(2, 3)
>>> F.zeros_like(x)
Tensor([[0 0 0]
 [0 0 0]], dtype=int32, device=xpux:0)