pytorch/tensor 操作

【pytorch】tensor的操作

• 矩阵转置操作 transpose与permute

  • • torch.transpose

    • • transpose函数的功能说明
      • 该函数的两种调用形式
      • 多次调用transpose实现多维张量的转置操作

    • torch.Tensor.permute

    • • permute函数的功能介绍

  • torch.Tensor.view

  • torch.Tensor.repeat

  • torch.cat

转置 transpose和permute

对于二维矩阵而言，转置操作意味着将行与列的位置进行交换，即原本表示为X[i][j]的元素，在转置之后将被表示为X[j][i]。

torch.transpose

transpose函数功能解析

torch.transpose(input, dim0, dim1) → Tensor

transpose函数的作用是将输