当前位置: 代码迷 >> 综合 >> PyTorch——torch.Tensor与np.ndarray(NumPy)之间的类型转换
  详细解决方案

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

热度:88   发布时间:2024-01-24 11:56:34.0

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