jax_sn.operators.scatter_operator.scatter_operator_strategies module¶
- class IndexedScatter[source]¶
Bases:
ScatterStrategy- forward(data: ScatterData, x)[source]¶
- total_scatter_kernel_xs_indexed(g_out: int, g_in: int, cross_sections_indexed: CrossSectionsIndexed, moments_array: Array)[source]¶
- Return type:
Array