PyTorch——torch.Tensor与np.ndarray（NumPy）之间的类型转换

1 前言

今天在写 Digit Recognizer的代码~

在对提交文件submission.csv进行写入操作的时候，总会有报错，其中一个原因是其实pandas对np.ndarray数据更加友好，所以这里我们就来记录一下torch.Tensor与np.ndarray（NumPy）之间进行数据类型转换的操作~

2 torch.Tensor与np.ndarray之间的类型转换

torch.Tensor to np.ndarray

使用如下的代码进行转换，

# 将 Tensor a转换成 ndarray b
b = a.numpy()