megengine.functional.zeros_like

zeros_like(inp)[源代码]

返回一个与输入张量形状相同的全0张量。

参数

inp – 输入张量。

返回

输出的全0张量。

例如:

import numpy as np
from megengine import tensor
import megengine.functional as F

inp = tensor(np.arange(1, 7, dtype=np.int32).reshape(2,3))
out = F.zeros_like(inp)
print(out.numpy())

输出:

[[0 0 0]
 [0 0 0]]