jax.experimental.sparse.JAXSparse

class jax.experimental.sparse.JAXSparse(args, *, shape)

高階 JAX 稀疏物件的基底類別。

參數:

• args (tuple[Array, ...])

• shape (tuple[int, ...])

__init__(args, *, shape)

參數:

• args (tuple[Array, ...])

• shape (Sequence[int])

方法

__init__(args, *, shape)
block_until_ready()
sum(*args, **kwargs)
transpose([axes])
tree_flatten()
tree_unflatten(aux_data, children)

屬性

T
ndim
size
data
shape
nse
dtype