jax.experimental.sparse.random_bcoo

jax.experimental.sparse.random_bcoo(key, shape, *, dtype=<class 'jax.numpy.float64'>, indices_dtype=None, nse=0.2, n_batch=0, n_dense=0, unique_indices=True, sorted_indices=False, generator=<function uniform>, **kwds)

生成一个随机的 BCOO 矩阵。

参数:

• key – 传递给 generator 函数的 PRNG 键。

• shape – 指定要生成的数组形状的元组。

• dtype – 要生成的数组的 dtype。

• indices_dtype – BCOO 索引的 dtype。

• nse – 矩阵中指定元素的数量，或者如果 0 < nse < 1，则为稀疏维度中指定的一部分（默认值：0.2）。

• n_batch – 批量维度数量。必须满足 n_batch >= 0 且 n_batch + n_dense <= len(shape)。

• n_dense – 批量维度的数量。必须满足 n_dense >= 0 且 n_batch + n_dense <= len(shape)。

• unique_indices – 指定索引是否应唯一的布尔值（默认：True）。

• sorted_indices – 指定索引是否应按字典顺序对行进行排序的布尔值（默认：False）。

• generator – 用于生成随机值的函数，接受键、形状和数据类型。它默认使用 jax.random.uniform()，并且可以是任何具有类似签名的函数。

• **kwds – 传递给 generator 的额外关键字参数。

返回:

具有指定属性的稀疏.BCOO数组。

返回类型:

arr