鉴于最近老是忘记torch,numpy,pandas之间的转换关系以及使用的注意事项,现在写一篇文章记录一下使用时候容易忘记的坑 torch在cuda和cpu下相同操作的不同函数 import torch data = torch.tensor([[1,2,3],[4,5,6]]) data.reshape(2,3) data = data.cuda() data.view(2,3) ou
实际上pandas的DataFrame先转换成np.array,再创建tensor #pandas和torch的转换 df=pd.DataFrame(np.arange(20).reshape(4,5),columns=['a','b','c','d','e'],index=['beijing','shanghai','wuhan','guangzhou']) t1=torch.from_numpy(df.values) #t1和df共享内存,实际上是DataFame->numpy.array->...