使用GPU进行训练&常见的Tensor&Hub模块
需要将数据和模型传入到CUDA里进行运算。
在代码上：
与传统的不同的是我们需要指定一个设备，这个设备就是设定为cuda:0
需要把模型放置到我们设定的设备中(CUDA中) .to(设备)
需要把数据放置到我呢设定的设备中(CUDA中） .to(设备)
常见的Tensor有哪些形式
| 形式 | 备注 | 代码&注释 |
|---|---|---|
| Scalar | 数值 | dim是0，标量 |
| Vector | 向量 | dim是1，在深度学习中通常指特征，例如词向量，某一个维度的特征等 |
| matrix | 矩阵 | dim是2，一行就是一个向量。可以认为每一行是每一个人，每一列是同一种特征。 矩阵能做乘法能做内积 |
| n-dimensal tensor | 高维张量 | Aijk… |
强大的Hub模块
Hub是为了方便开发者调用已有的代码,是一个model zoo
torch.hub.load()
学习一般不用这个