在PyTorch中創建張量有多種方法,最常用的方法包括:
import torch
tensor = torch.tensor([1, 2, 3, 4, 5])
import torch
ones_tensor = torch.ones(2, 3)
zeros_tensor = torch.zeros(3, 2)
import torch
random_tensor = torch.randn(2, 3)
import torch
arange_tensor = torch.arange(1, 10, 2)
linspace_tensor = torch.linspace(1, 10, 5)
除了以上方法外,還可以根據自己的需求使用其他方法來創建張量,如torch.eye()函數用于創建單位矩陣張量等。