PyTorch 与神经网络学习

PyTorch 的基础操作

1 张量

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

1.1 张量初始化与创建

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

版权声明:本文为qq_43961980原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。