今天开始入坑深度学习,首先学习难度较低的pytorch框架。这一篇文章就是简单学习下pytorch的基本操作。
1. pytorch数据类型
pytorch中的数据类型为tensor,它的基本操作和numpy都挺相似的。要注意的有两个地方,要查看数据的形状不再是shape(),而是view();改变形状不再是reshape(),而是view()。
import torch
import numpy as np
torch.tensor([2,3])
torch.tensor([[2,3],[4,5]