jax.experimental.sparse.coo_matvec

目录

jax.experimental.sparse.coo_matvec#

jax.experimental.sparse.coo_matvec(mat, v, transpose=False)[源代码][源代码]#

COO 稀疏矩阵与密集向量的乘积。

参数:
  • mat (COO) – COO 矩阵

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

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

返回:

形状为 (mat.shape[1] if transpose else mat.shape[0],) 的数组,表示矩阵向量乘积。

返回类型:

y