Rearrange

Rearrange#

class seli.net.Rearrange(pattern: str, **dims: int)[source]#

Bases: Module

Rearrange the input tensor according to the given pattern. See einops.rearrange for more information.

Parameters:

  • pattern (str) – The pattern to rearrange the input tensor.

  • dims (int) – The dimensions to use in the pattern.

Methods Summary

__call__(*args)

Call self as a function.

Methods Documentation

__call__(*args)[source]#

Call self as a function.

Contents