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()