jax.experimental.sparse.csr_matmat# jax.experimental.sparse.csr_matmat(mat, B, *, transpose=False)[源代码][源代码]# 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
