sheaf_mpnn.nsd.NSDVariant#

class NSDVariant[source]#

Bases: Enum

__init__(*args, **kwds)#

Methods

build_kwargs([orth_strategy, rank])

Build the full layer keyword-argument dict for this variant.

Attributes

`layer_class`
`layer_kwargs`
`DIAGONAL`
`GENERAL`
`ORTHOGONAL`
`GENERAL_ATTENTION`
`ORTHOGONAL_ATTENTION`
`LOW_RANK`

DIAGONAL = 1#

GENERAL = 2#

ORTHOGONAL = 3#

GENERAL_ATTENTION = 4#

ORTHOGONAL_ATTENTION = 5#

LOW_RANK = 6#

property layer_class#

property layer_kwargs: dict[str, Any]#

build_kwargs(orth_strategy='cayley', rank=1)[source]#

Build the full layer keyword-argument dict for this variant.

Parameters:

orth_strategy (Literal['cayley', 'fasth'], default: "cayley")
rank (int, default: 1)

Return type:

dict[str, Any]