服务器系列（23）：tensor、numpy.array、list三者之间互相转换

Posted 2023-04-26

参考技术A ValueError:only one element tensors can be converted to Python scalars问题解答
1.1 list 转 numpy

ndarray = np.array(list)

1.2 numpy 转 list

list = ndarray.tolist()

2.1 list 转 torch.Tensor

tensor=torch.Tensor(list)

2.2 torch.Tensor 转 list

先转numpy，后转list

list = tensor.numpy().tolist()

3.1 torch.Tensor 转 numpy

ndarray = tensor.numpy()

*gpu上的tensor不能直接转为numpy

ndarray = tensor.cpu().numpy()

3.2 numpy 转 torch.Tensor

tensor = torch.from_numpy(ndarray)

张量系列-Tensor(01)

张量——N-dim 数组

1. 数组的创建

2. 符号数组的创建

3. 一维数组改变形状创建

4. 切片操作

5. 符号数组操作

6. 数组转化为列表

7. 维度为2的数组可以转化为矩阵