jax.experimental.sparse.bcoo_dot_general#

jax.experimental.sparse.bcoo_dot_general(lhs, rhs, *, dimension_numbers, precision=None, preferred_element_type=None, out_sharding=None)[來源]#

通用的收縮操作。

參數:

  • lhs (BCOO | Array) – ndarray 或 BCOO 格式的稀疏陣列。

  • rhs (BCOO | Array) – ndarray 或 BCOO 格式的稀疏陣列。

  • dimension_numbers (DotDimensionNumbers) – 形式為 ((lhs_contracting_dims, rhs_contracting_dims), (lhs_batch_dims, rhs_batch_dims)) 的元組。

  • precision (None | None) – 未使用

  • preferred_element_type (None | None) – 未使用

傳回:

包含結果的 ndarray 或 BCOO 格式的稀疏陣列。如果兩個輸入都是稀疏的，則結果將為稀疏的，類型為 BCOO。如果任一輸入是密集的，則結果將為密集的，類型為 ndarray。

傳回類型:

BCOO | Array