core.models.utils.irreps

core.models.utils.irreps#

Copyright (c) Meta Platforms, Inc. and affiliates.

This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

`cg_change_mat`(→ torch.tensor)
`irreps_sum`(→ int)	Returns the sum of the dimensions of the irreps up to the specified angular momentum.

core.models.utils.irreps.cg_change_mat(ang_mom: int, device: str = 'cpu') → torch.tensor#

core.models.utils.irreps.irreps_sum(ang_mom: int) → int#

Returns the sum of the dimensions of the irreps up to the specified angular momentum.

Parameters:: ang_mom – max angular momenttum to sum up dimensions of irreps