溫馨提示×

Torch中怎么創(chuàng)建一個張量

小億
119
2024-03-18 15:29:46

在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的張量。

0