jax.experimental.sparse.csr_matvec

jax.experimental.sparse.csr_matvec(mat, v, transpose=False)

CSR 稀疏矩阵和稠密向量的乘积。

参数：

• mat (CSR) – CSR 矩阵

• v (Array) – 大小为 (shape[0] if transpose else shape[1],) 且数据类型为 mat.dtype 的一维数组

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

返回值：

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

矩阵向量积。

返回类型：

y