pytorch处理二维tensor

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)一样

---

numpy中操作

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

（3）、numpy转tensor

import numpy as np
import torch
a = np.ones(5)
b = torch.from_numpy(a)
print(a, b)
a += 1
print(a, b)
b += 1
print(a, b)

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)

---

一个矩阵的第一行＋另一个矩阵第一行
但要保证其中一个是只有一个元素

---