src.model.permuted module

class src.model.permuted.Permuted(*args: Any, **kwargs: Any)

Bases: Module

Dimension permutation layer.

Wraps Tensor.permute so it can be used inside nn.Sequential.

__init__(*dims)

Parameters:

*dims (int) – Target dimension ordering.

forward(x)