提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
python笔记3:numpy.array、list、torch.tensor互换
一、list与numpy.array数组互换
1.list 向numpy.array数组转换
numpy.array(list),参数为列表类型
一维转换
二维转换
2.numpy.array向list 数组转换
ndarray.tolist(),变量名在前,数据类型是ndarray
二、list与torch.tensor数组互换
1.list向torch.tensor数组转换
torch.tensor(list),参数为列表类型
2.torch.tensor向list数组转换
tensor.tolist(),变量名在前,数据类型是torch.tensor
- torch.tensor与list互换,维度并不会影响
torch.range与torch.arange创建出来的tensor,数据类型不一样,参数也不一样。range创建时,可输入起末位置,返回值并包含末位置,且是torch.float32型数据;arange创建时,只输入末位置,默认从零开始,返回值不包含末位置,数据类型是torch.int64.