jax.experimental.sparse.COO

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

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

注意：此类与 JAX 变换（如 grad 和自动微分）的兼容性很低，并且提供的功能很少。通常情况下，您应该优先使用 jax.experimental.sparse.BCOO.

此外，如果 `nse` 大于表示矩阵中非零元素的真实数量，则会出现已知的错误。这种情况在 BCOO 中得到更好的处理。

参数：

• args (元组[数组, 数组, 数组])

• shape (元组[整数, 整数])

• rows_sorted (布尔值)

• cols_sorted (布尔值)

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

参数：

• args (元组[数组, 数组, 数组])

• shape (形状)

• rows_sorted (布尔值)

• cols_sorted (布尔值)

方法

__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
row
col
shape