megengine.functional.zeros

zeros(shape, *, dtype='float32', device=None)[源代码]

返回一个特定形状的全0的新tensor。

参数

shape (int or sequence of ints) – 输出tensor的shape。

关键字参数
  • dtype (Tensor.dtype) – 输出tensor的数据类型。默认值:float32

  • device (Tensor.device) – 放置被创建tensor的设备。默认值: None

返回类型

Tensor

返回

一个包含0的tensor。

实际案例

>>> F.zeros((2, 1))
Tensor([[0.]
 [0.]], device=xpux:0)