在PyTorch中,可以使用torch.tensor()
函數來創建張量。例如,創建一個包含5個元素的一維張量可以這樣做:
import torch
tensor = torch.tensor([1, 2, 3, 4, 5])
print(tensor)
這將會輸出類似以下的結果:
tensor([1, 2, 3, 4, 5])
另外,PyTorch還提供了一些其他函數來創建特定類型的張量,比如torch.zeros()
可以創建全零張量,torch.ones()
可以創建全一張量,torch.randn()
可以創建從標準正態分布中抽取的隨機張量等等。