irreps_zeros_like

cuequivariance_jax.ir_dict.irreps_zeros_like(x)

Create a dict[Irrep, Array] of zeros with the same structure.

Parameters:

x (dict[Irrep, Array]) – Template dictionary with arrays of shape (…, mul, ir.dim).

Returns:

Dictionary with zero arrays of the same shapes and dtypes.

Return type:

dict[Irrep, Array]