当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch zeros()用法及代码示例


PyTorch是由Facebook开发的开源机器学习库。它用于深度神经网络和自然语言处理。

函数torch.zeros()返回一个由标量值0填充的张量,其形状由变量参数size定义。

用法:torch.zeros(size, out=None)

参数
size:定义输出张量形状的整数序列
out (Tensor, optional):输出张量

返回类型:一个张量,其标量值为0,形状与尺寸相同。



代码1:

# Importing the PyTorch library 
import torch 
  
  
# Applying the zeros function and 
# storing the resulting tensor in 't' 
a = torch.zeros([3, 4]) 
print("a = ", a) 
  
b = torch.zeros([1, 5]) 
print("b = ", b) 
  
c = torch.zeros([5, 1]) 
print("c = ", c) 
  
d = torch.zeros([3, 3, 2]) 
print("d = ", d)

输出:

a =  tensor([[0., 0., 0., 0.],
        [0., 0., 0., 0.],
        [0., 0., 0., 0.]])
b =  tensor([[0., 0., 0., 0., 0.]])
c =  tensor([[0.],
        [0.],
        [0.],
        [0.],
        [0.]])
d =  tensor([[[0., 0.],
         [0., 0.],
         [0., 0.]],

        [[0., 0.],
         [0., 0.],
         [0., 0.]],

        [[0., 0.],
         [0., 0.],
         [0., 0.]]])

相关用法


注:本文由纯净天空筛选整理自sanskar27jain大神的英文原创作品 Python PyTorch zeros() method。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。