溫馨提示×

PyTorch中怎么創(chuàng)建張量

小億
83
2024-03-16 15:54:46
欄目: 深度學習

在PyTorch中,可以使用torch.tensor()函數來創(chuàng)建張量。例如,創(chuàng)建一個包含5個元素的一維張量可以這樣做:

import torch

tensor = torch.tensor([1, 2, 3, 4, 5])
print(tensor)

這將會輸出類似以下的結果:

tensor([1, 2, 3, 4, 5])

另外,PyTorch還提供了一些其他函數來創(chuàng)建特定類型的張量,比如torch.zeros()可以創(chuàng)建全零張量,torch.ones()可以創(chuàng)建全一張量,torch.randn()可以創(chuàng)建從標準正態(tài)分布中抽取的隨機張量等等。

0