Reduce

Reduce#

class seli.net.Reduce(pattern: str, reduction: Literal['sum', 'mean', 'max', 'min', 'prod'], **dims)[source]#

Bases: Module

Reduce the input tensor according to the given pattern and the reduction type. See einops.reduce for more information.

Parameters:

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

  • reduction (str) – The reduction type to use. Can be one of “sum”, “mean”, “max”, “min”, or “prod”.

  • 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