jax.experimental.sparse.csr_matmat# jax.experimental.sparse.csr_matmat(mat, B, *, transpose=False)[source]# CSR 稀疏矩阵和密集矩阵的乘积。 参数: mat (CSR) – CSR 矩阵 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
