【Pytorch教程】:Torch and Numpy
1. 自由地转换 numpy array 和 torch tensorimport torchimport numpy as np# convert numpy to tensor or vise versanp_data = np.arange(6).reshape((2, 3))torch_data = torch.from_numpy(np_data)tensor2array = torch_data.numpy()print( '\nnumpy array:', np_d
复制链接