一、生成array\list\tensor
1.生成array
import numpy as np
a1 = array([[1,2],[3,4]])
2. 生成list
a2 = list(range(16))
3.生成tensor
import torch
a3 = torch.tensor([[1,2],[3,4]])
还有一些生成特定的tensor方法,比如：
torch.zeros(3) #生成3*3的zero-tensor
torch.zeros_like(tensora) #生成一个与tensora维度相同的zero-tensor
torch.ones(3) #生成3*3的one-tensor
torch.ones_like(tensora) #生成一个与tensora维度相同的one-tensor
torch.full(dims, value) #生成全部是value的dims维tensor
torch.eye(dim) #单元tensor eye只能建立方阵
二、相互转换
1.list 与 array相互转换
注意到array是numpy中的。因此导入numpy包。利用np.array()和 a.tolist()来实现转换。
a1 = np.array([[1,2],[3,4]])
m = a1.tolist() # array 2 list
m.remove(m[0]) #进行一些操作
a2 = np.array(m) #list 2 array
2. list 与 tensor 相互转换
t = torch.Tensor(l) # list 2 tensor
l = t.numpy().tolist() # tensor 2 list
3.array 与 tensor 相互转换
t = torch.from_numpy(n) # numpy 2 tensor
n = t.numpy() # tensor 2 numpy
三、不同格式数据的类型转换
t.int() # tensor
n.astype(np.int32) # array
l = [int(i) for i in l] # list
l = list(map(int, l)) # list
ref: csdn_blog
