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

PyTorch是由Facebook開發的開源機器學習庫。它用於深度神經網絡和自然語言處理。

函數torch.zeros()返回一個由標量值0填充的張量，其形狀由變量參數size定義。

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

代碼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.]]])