(注意:代码未导入相关包和进行初始化赋值不能直接运行)
一、list和numpy之间的转换(np表示numpy对象,lists表示list对象)
np = numpy.array(lists) # list转numpy lists = np.tolist() # numpy转list
二、numpy和tensor之间的转换(t表示tensor对象,np表示numpy对象)
t = torch.from_numpy(np) # numpy转tensor一种方式 t = torch.Tensor(np) # numpy转tensor另一种方式 np = t.numpy() # tensor转numpy
三、list和tensor之间的转换(t表示tensor对象,lists表示list对象)
t = torch.Tensor(lists) # list转tensor lists = t.tolist() # tensor转list lists = t.numpy().tolist() # tensor先转numpy再转list