xnn.gnn.models.blocks.species_irreps#

xnn.gnn.models.blocks.species_irreps(n_species)[source]#

Irreps of the one-hot species (node attribute) channels.

Parameters:: n_species (int) – Number of distinct chemical species.
Returns:: n_species even scalar irreps, i.e. o3.Irreps([(n_species, (0, 1))]) (written {n_species}x0e).
Return type:: o3.Irreps