Python Pytorch full()用法及代碼示例

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

函數torch.full()返回一個大小為fill_value的張量的張量。

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

代碼1：

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

輸出：

a =  tensor([[3., 3., 3., 3.],
        [3., 3., 3., 3.],
        [3., 3., 3., 3.]])
b =  tensor([[3.5000, 3.5000, 3.5000, 3.5000, 3.5000],
        [3.5000, 3.5000, 3.5000, 3.5000, 3.5000]])