PyTorch基础知识笔记(1)
PyTorch是一个动态的建图的工具。不像Tensorflow那样,先建图,然后通过feed和run重复执行建好的图。相对来说,PyTorch具有更好的灵活性。
参数
pytorch中有两种变量类型,一个是Tensor,一个是Variable。
Tensor: 就像ndarray一样,一维Tensor叫Vector,二维Tensor叫Matrix,三维及以上称为Tensor
Variable:是Tensor的一个wrapper,不仅保存了值,而且保存了这个值的creator,需要BP的网络都是Variab
原创
2020-05-28 16:17:43 ·
246 阅读 ·
0 评论