jax.experimental.sparse.CSC

jax.experimental.sparse.CSC#

class jax.experimental.sparse.CSC(args, *, shape)[源代码][源代码]#

在JAX中实现的实验性CSC矩阵;API可能会有变动。

参数:

shape (tuple[int, int])

__init__(args, *, shape)[源代码][源代码]#

方法

__init__(args, *, shape)

block_until_ready()

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

sum(*args, **kwargs)

todense()

transpose([axes])

tree_flatten()

tree_unflatten(aux_data, children)

属性

T

dtype

ndim

nse

size

data

indices

indptr

shape