jax.experimental.sparse.COO

class jax.experimental.sparse.COO(args, *, shape, rows_sorted=False, cols_sorted=False)

JAX 中实现的实验性 COO 矩阵。

注意：此类与 JAX 变换（如 grad 和 autodiff）的兼容性最低，功能也很少。通常应优先使用 jax.experimental.sparse.BCOO。

此外，当 nse 大于所表示矩阵的实际非零元素数量时，存在已知的故障。BCOO 在此情况下处理得更好。

参数:

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

• shape (tuple[int, int])

• rows_sorted (bool)

• cols_sorted (bool)

__init__(args, *, shape, rows_sorted=False, cols_sorted=False)

参数:

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

• shape (Shape)

• rows_sorted (bool)

• cols_sorted (bool)

方法

__init__(args, *, shape[, rows_sorted, ...])
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
行
列
shape