from Dzwfvz
import torch
import numpy as np
np_data=np.arange(6).reshape((2,3)) # numpy形式
torch_data=torch.from_numpy(np_data) # tensor向量
tensor2array=torch_data.numpy() # array形式
成功解决：numpy形式转换为tensor向量，tensor向量转换为array形式
Guess you like
Origin blog.csdn.net/shine_00/article/details/121431878
Recommended
Ranking