Einsum

class seli.net.Einsum(pattern: str)

Bases: Module

Wrapper around einops.einsum. Performs a contraction on the input tensors according to the given pattern.

Parameters:

• pattern (str) – The pattern to contract the input tensors.

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

Methods Summary

__call__(*args)

Call self as a function.

Methods Documentation

__call__(*args)

Call self as a function.