Pytorch实践中的list、numpy、torch.tensor之间数据格式的相互转换方法-阿里云开发者社区

link之家

链接快照平台

相关文章推荐

咆哮的牛肉面 · tmux：Linux终端的强大会话管理神器- ...· 5 月前 ·

仗义的沙发 · R语言时间序列基础库zoo_r ...· 2 年前 ·

正直的皮蛋 · 13行python代码实现对微信进行推送消息 ...· 2 年前 ·

怕老婆的水龙头 · WPF ...· 2 年前 ·

（注意：代码未导入相关包和进行初始化赋值不能直接运行）

一、list和  numpy 之间的转换（np表示numpy对象，lists表示list对象）

np = numpy.array(lists) # list转numpy
lists = np.tolist() # numpy转list

t = torch.from_numpy(np) # numpy转tensor一种方式
t = torch.Tensor(np) # numpy转tensor另一种方式
np = t.numpy() # tensor转numpy

t = torch.Tensor(lists) # list转tensor
lists = t.tolist() # tensor转list
lists = t.numpy().tolist() # tensor先转numpy再转list