在Torch中,可以使用torch.tensor()函數來創建一個張量。例如:
torch.tensor()
import torch # 創建一個大小為3x3的隨機張量 tensor = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) print(tensor)
這樣就創建了一個3x3的隨機張量。也可以使用torch.zeros()、torch.ones()等函數來創建全0或全1的張量。
torch.zeros()
torch.ones()
億速云公眾號
手機網站二維碼
Copyright ? Yisu Cloud Ltd. All Rights Reserved. 2018 版權所有
廣州億速云計算有限公司粵ICP備17096448號-1 粵公網安備 44010402001142號增值電信業務經營許可證編號:B1-20181529