megengine.functional.zeros_like

zeros_like(inp)

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

参数

inp (Tensor) – 输入张量。

返回类型

Union[Tensor, SymbolVar]

返回

一个全零的tensor。

实际案例

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

上一页

megengine.functional.zeros

下一页

megengine.functional.ones