peal.architectures.basic_modules.Transpose

class peal.architectures.basic_modules.Transpose[source]

Bases: Module

Module form of x.transpose(dim1, dim2).

Parameters:
  • dim1 (int) – Dimensions to swap.

  • dim2 (int) – Dimensions to swap.

Methods

forward

Swap the two configured dimensions of x.

__init__(dim1, dim2)[source]
forward(x)[source]

Swap the two configured dimensions of x.