在Torch中,可以使用torch.tensor()
函數(shù)來創(chuàng)建一個張量。例如:
import torch
# 創(chuàng)建一個大小為3x3的隨機張量
tensor = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(tensor)
這樣就創(chuàng)建了一個3x3的隨機張量。也可以使用torch.zeros()
、torch.ones()
等函數(shù)來創(chuàng)建全0或全1的張量。