PyTorch和Numpy之间的联系和差异的详细阐述(万文多图)

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

import numpy as np
n = np.random.