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],) 且 dtype 為 mat.dtype 的一維陣列

• transpose (bool) – 布林值，指定是否在計算之前轉置稀疏矩陣。

傳回值:

形狀為 (mat.shape[1] if transpose else mat.shape[0],) 的陣列，表示

矩陣向量乘積。

傳回型別:

y