(L, axis=0)
| 176 | ) from e |
| 177 | |
| 178 | def _concat_cupy_sparse(L, axis=0): |
| 179 | if axis == 0: |
| 180 | return vstack(L) |
| 181 | elif axis == 1: |
| 182 | return hstack(L) |
| 183 | else: |
| 184 | msg = ( |
| 185 | "Can only concatenate cupy sparse matrices for axis in " |
| 186 | f"{{0, 1}}. Got {axis}" |
| 187 | ) |
| 188 | raise ValueError(msg) |
| 189 | |
| 190 | concatenate_lookup.register(spmatrix, _concat_cupy_sparse) |
| 191 | tensordot_lookup.register(spmatrix, _tensordot_scipy_sparse) |