jax.experimental.sparse.coo_matmat

jax.experimental.sparse.coo_matmat(mat, B, *, transpose=False)

COO 稀疏矩阵和稠密矩阵的乘积。

参数:

• mat (COO) – COO 矩阵

• B (Array) – 形状为 (mat.shape[0] if transpose else mat.shape[1], cols)，数据类型为 mat.dtype 的数组。

• transpose (bool) – 一个布尔值，指定是否在计算之前转置稀疏矩阵。

返回:

形状为 (mat.shape[1] if transpose else mat.shape[0], cols) 的数组

表示矩阵向量积。

返回类型:

C