pytorch报错及问题解决_记录1
1. tensor转array
要记得在numpy后加括号!
Img_array = Img_tensor.numpy()
2. array转list
Img_list = Img_array.tolist()
3. list拼接:
patterns_sum.append(Img_list)
4. list转tensor:
- 一般list 转tensor只需要
patterns_sum_tensor = torch.tensor(patterns_sum)
- 但是要转换的list里面的元素包含多维的tensor
x_DGI_tensor = torch.stack(x_DGI)
5. tensor拼接思路:
tensor转array,array转list,list拼接,list转tensor
6. 字符串list转数字list:
s_number = [float(x) for x in s_str]
7. tensor切片
[start : end : step]
ss = s_tensor[1:301:1]
8. tensor点对点相加
x_DGI = dgi_750 + dgi_1500
9. tensor用img读取并展示
思路:tensor转numpy,再用plt读取
x_DGI_numpy = x_DGI_r.numpy()
plt.imshow(x_DGI_numpy)