megengine.functional.zeros_like¶

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

参数: inp (Union[Tensor, SymbolVar]) – 输入张量。
返回类型: Union[Tensor, SymbolVar]
返回: output tensor.

实际案例

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

megengine.functional.zeros

megengine.functional.ones