jax.experimental.sparse.CSC

Contents

jax.experimental.sparse.CSC#

class jax.experimental.sparse.CSC(args, *, shape)[source]#

Experimental CSC matrix implemented in JAX; API subject to change.

__init__(args, *, shape)[source]#

Methods

__init__(args, *, shape)

block_until_ready()

fromdense(mat, *[, nse, index_dtype])

sum(*args, **kwargs)

todense()

transpose([axes])

tree_flatten()

tree_unflatten(aux_data, children)

Attributes

T

dtype

ndim

nse

size

data

indices

indptr

shape