e3x.nn.wrappers.AngularFn

class e3x.nn.wrappers.AngularFn(*args, **kwargs)[source]

Bases: Protocol

Protocol for angular functions.

__call__(r, max_degree, cartesian_order)[source]

Return type:: <class 'Float[Array, '... (max_degree+1)**2']'>