jax.experimental.sparse.coo_matmat

jax.experimental.sparse.coo_matmat(mat, B, *, transpose=False)

COO 稀疏矩陣與稠密矩陣的乘積。

參數:

• mat (COO) – COO 矩陣

• B (Array) – 形狀為 (mat.shape[0] if transpose else mat.shape[1], cols) 且 dtype 為 mat.dtype 的陣列

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

回傳:

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

表示矩陣向量乘積。

回傳型別:

C