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) – 一个 n维数组或 BCOO 格式的稀疏数组。

  • rhs (BCOO | Array) – 一个 n维数组或 BCOO 格式的稀疏数组。

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

  • precision (None) – 未使用

  • preferred_element_type (None) – 未使用

返回:

一个包含结果的 n维数组或 BCOO 格式的稀疏数组。如果两个输入都为稀疏，则结果将为稀疏，类型为 BCOO。如果任一输入为密集，则结果将为密集，类型为 n维数组。

返回类型:

BCOO | Array