pytorch最全学习笔记

pytorch安装教程

见: 安装教程
安装 PyTorch 会安装两个模块，一个是torch，一个 torchvision, torch 是主模块，用来搭建神经网络的，torchvision 是辅模块，有数据库，还有一些已经训练好的神经网络等着你直接用，比如 (VGG, AlexNet, ResNet)。

torch与numpy

torch_data = torch.from_numpy(np_data)可以将numpy(array)格式转换为torch(tensor)格式；
torch_data.numpy()又可以将torch的tensor格式转换为numpy的array格式。
注意Torch的Tensor和numpy的array会共享他们的存储空间，修改一个会导致另外的一个也被修改。

1.torch和numpy的转换

import torch
import numpy as np

np_data = np.arange(6).reshape((2, 3))
torch_data = torch.from_numpy(np_data)
tensor2array = torch_data.numpy()