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形式
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形式