jax.experimental.sparse.bcoo_dot_general_sampled

jax.experimental.sparse.bcoo_dot_general_sampled#

jax.experimental.sparse.bcoo_dot_general_sampled(A, B, indices, *, dimension_numbers)[source]#

A contraction operation with output computed at given sparse indices.

Parameters:
Return type:

Array

Returns:

BCOO data, an ndarray containing the result.