#int型tensor

from torch.autograd import Variable

from torch import IntTensor

var = Variable(IntTensor([[1,0],[0,1]]))

#檢視size

var.size()   #torch.Size([2, 2])

#將var.size()轉換為list

list(var.size())  #[2,2]

#檢視元素總個數

z.numel()  

#torch 增加維度

x = torch.Tensor(2,4).zero_()    #2*4維的全0 tensor

y = x.unsqueeze(0)  #y的維度為1，2，4

y = x.unsqueeze(1) #y的維度為 2，1，4

y = x.unsqueeze(2) #y的維度為2，4，1

#view()函式的使用

a = torch.Tensor(2,3)

a.view(1,-1)  #變成1*6維

a.view(6,1) #變成6*1維

a.view(3,2) #變成3*2維

#矩陣相乘

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

torch.mm(tensor,tensor)  #結果為 tensor([[7.,10.],[15.,22.]])

#點乘

tensor_a.dot(tensor_b) #[1,2,3,4] [1,2,3,5]  輸出為34