PyTorch tensor operations reference (速查表)

PyTorch张量常用的创建、变形及数学运算总结

目录

• PyTorch中张量的常见创建方式、变形操作及数值运算汇总

  • • 1. 张量的生成
    • • 1.1 生成张量的函数
      • 生成不同类型的三维数组：torch.Tensor() 和 torch.tensor()
      • 创建特定类型的数组：torch.rand()（随机数）、torch.ones()（全1数组）、torch.zeros()（全0数组）
      • 初始化不同存储格式的数据结构：
        • torch.FloatTensor()（浮点型）
        • torch.DoubleTensor()（双精度浮点型）
        • torch.cuda.FloatTensor()（浮点型且在GPU上）
        • 类似的方法用于其他数据类型和存储环境

• 基本运算与操作

  • 确定张量形状
    • 执行变形操作：视图(view())、转置(transpose())以及展平(flatten())
    • 实施基本算术运算：加法(+), 减法(-), 乘法(*), 零填充(\0)，并进行矩阵相乘(@)和全零填充(matnul)
    • 进行分割与拆解(split, chunk)，合并与连接(