Repeat

Repeat#

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

Bases: Module

Repeat the input tensor according to the given pattern. See einops.repeat for more information.

Parameters:

  • pattern (str) – The pattern to repeat 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