sparse_astype

hypercoil.functional.sparse.sparse_astype(tensor: Tensor, dtype: Any) → Tensor

Set the data type of a sparse matrix.

This function is probably unnecessary, but I’m missing a way to do this with the current JAX API.