Crate caffe2op_transpose

source ·

Structs

  • CudnnTransposeOp
  • GetTransposeGradient
  • TransposeOp
    | Transpose the input tensor by permuting | the axes of the input according to the | axes argument. | | Similar to numpy’s transpose | function. | | For example, when axes=(1, 0, 2), given | an input tensor of shape (1, 2, 3), the | output shape will be (2, 1, 3). | | Github Links: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/transpose_op.cc |