文章目录
0. PyTorch介绍
A Tensor library like Numpy, unlike Numpy it has strong GPU support. Lua is a wrapper for Torch (Yes! you need to have a good understanding of Lua), and for that you will need LuaRocks package manager.
由此可见,PyTorch和Numpy有着千丝万缕的关系。为了提高学习效率,特意把两者之间的关联总结出来。
1. Pytorch数据
1.1 如何获取数据的基本属性
数据的基本属性包括数据的维度、元素个数、每一维的大小。
- Numpy
基本属性 | Numpy类别 | API |
---|---|---|
维度 | 属性 | ndim |
元素个数 | 属性 | size |
每一维的大小 | 属性 | shape |