技術標籤：PyTorchPythonnumpypython

1、建立tensor

(1)、torch.arange建立tensor

t = torch.arange(0,10)
print(t)

左閉右開

（2）、一維tensor轉二維矩陣

t = torch.arange(0,10).view(2,5)
print(t)
print(t.shape)

view(2,5)是將一維tensor轉為2*5的二維tensor，總體元素個數是相同的

t = torch.arange(0,20).view(2,2,5)
print(t)

此處便將20個元素轉為三維tensor 2*(2*5)

view()與resize_、reshape得出結果相同

x = torch.arange(0,15).resize_(3,5)
y = torch.arange(0,3).resize_(3,1)
print(x)
print(y)
print(x+y)
#和torch.add(x,y)一樣

（3）、numpy轉tensor

numpy中操作

y = np.arange(0,16).view()
print(y)
x = y.reshape(2,8)
print(x)

2、矩陣相加

t = torch.arange(0,10).view(2,5)
print(t)
print(t+t)
 

x = torch.arange(0,15).view(3,5)
y = torch.arange(0,3).view(3,1)
print(x)
print(y)
print(x+y)

一個矩陣的第一行＋另一個矩陣第一行
但要保證其中一個是隻有一個元素