indexed_linear

cuequivariance_jax.experimental.indexed_linear(

poly,

counts,

w,

x,

math_dtype=None,

method='indexed_linear',

)

Parameters:

• poly (SegmentedPolynomial)

• counts (Array)

• w (Array)

• x (Array)

• math_dtype (str | None)

• method (str)

Return type:

Array