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
。
- 返回类型
- 返回
一个包含0的tensor。
实际案例
>>> F.zeros((2, 1)) Tensor([[0.] [0.]], device=xpux:0)