1.元素的删除
list删除指定索引元素
a=[]
a.pop[index]
2.list/np/tensor之间的相互转换
(1)list转np
import numpy as np
a_ndarray = np.array(a_list)
(2)list转tensor
A_tensor = torch.Tensor(a_list)
(3)np转list
A_list = a_ndarray.tolist() # ndarray 转为 list
(4)np转tenor
a_tensor = torch.from_numpy(a_ndarray)
(5)tensor转np
# CPU
ndarray = tensor.numpy()
# GPU
ndarray = tensor.cpu().numpy()