torch和numpy的相互轉換

"""
pytorch相當於神經網絡中的numpy，它是以tensor的形式表示
"""
import torch
import numpy as np

# convert numpy to tensor or vise versa
np_data = np.arange(6).reshape((2, 3))
torch_data = torch.from_numpy(np_data) #numpy轉換為torch
tensor2array = torch_data.numpy() #torch轉換為numpy
print(
    '\nnumpy array:', np_data,
    '\ntorch tensor:', torch_data,
    '\ntensor to array:', tensor2array,
)