【pytorch学习笔记2021.10.20】numpy与torch相互转换
numpy->torch
pt_data = torch.from_numpy(np_data)
torch->numpy
np_data = pt_data.numpy()
实例:
import torch
import numpy as np
np_data = np.arange(6).reshape((2, 3))
torch_data = torch.from_numpy(np_data)
np_data_2 = torch_data.numpy()
pr..
原创
2021-10-20 10:26:17 ·
242 阅读 ·
0 评论