PyTorch 与神经网络学习

PyTorch 的基础操作

1 张量

  • 张量如同数组和矩阵一样,即一种特殊的数据结构。
  • 多作为 pytorch 中,神经网络的输入、输出以及网格的参数等数据,都用张量来描述
  • 张量的使用 和 numpy 的 ndarrays 类似,区别在于张量可以在GPU或者其他专用硬件上运行,以达到更快的效果

1.1 张量初始化与创建

# 初始化张量
'''
张量与numpy的数组最大的区别在于张量可以在gpu上运行
'''
data = [[1,2],[3,4

你可能感兴趣的:(机器学习,pytorch,神经网络)