xnn.gnn.models.blocks.hidden_irreps

xnn.gnn.models.blocks.hidden_irreps(mul, l_max)

Hidden feature irreps: mul copies of the spherical-harmonic irreps.

Parameters:

• mul (int) – Multiplicity (number of channels) per irrep degree.

• l_max (int) – Maximum degree l; the degrees/parities follow o3.Irreps.spherical_harmonics(l_max) (i.e. 0e, 1o, 2e, ...).

Returns:

mul copies of each spherical-harmonic irrep up to l_max.

Return type:

o3.Irreps