pytorch知识点小笔记

tensor嵌套tensor，可以理解为外tensor是菜单，内tensor是点的菜。

♪(･ω･)ﾉ

========================================

torch.tensor([5]) 与 torch.tensor(5) 不一样，哭了。折磨了半天

========================================

关于str.join()的用法

==============================================

关于数据从gpu转到cpu
#将GPU上的张量转换为cpu上的数值
#pred.cuda().data.cpu().numpy()

#获取变量的设备信息

device = torch.device(torch.get_device(x))

==============================================

pytorch中矩阵乘法（2,1,3）*(2,3,3)

思路：先将左矩阵，(2,1,3)自我复制为（2,3,3），然后左右矩阵对应位置相乘。

pytorch中矩阵乘法（2,1,1）*(2,3,3)

=================================================

numpy与torch互转

pytorch

import numpy as np

import torch

np_arr = np.array([1,2,3,4])

tor_arr=torch.from_numpy(np_arr)

tor2numpy=tor_arr.numpy()