pytorch初始化矩陣

import torch

a = torch.ones(5,3)     #5*3的元素全為1的矩陣
print(a)
b = torch.rand((5,3))   #5*3的元素為0-1之間的隨機數的矩陣
print(b)
c = torch.tensor([[1,2,3],
                  [4,5,6],
                  [7,8,9]])
print(c)
d = torch.zeros(5,3, dtype=torch.long)  #5*3的元素全為0的矩陣,dtype可以指定數據類型
print(d)
print(a.type()) #輸出為torch.FloatTensor
print(b.type()) #輸出為torch.FloatTensor
print(c.type()) #輸出為torch.LongTensor
print(d.type()) #輸出為torch.LongTensor