jax.experimental.sparse.bcoo_dot_general#
- jax.experimental.sparse.bcoo_dot_general(lhs, rhs, *, dimension_numbers, precision=None, preferred_element_type=None)[source]#
一个通用的收缩操作。
- 参数:
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。
- 返回类型: