jax.experimental.sparse.empty#

jax.experimental.sparse.empty(shape, dtype=None, index_dtype='int32', sparse_format='bcoo', **kwds)[source]#

创建一个空的稀疏数组。

参数:

  • shape (Shape) – 整数序列，表示数组的形状。

  • dtype (DTypeLike | None) – （可选）数组的数据类型。

  • index_dtype (DTypeLike) – （可选）索引数组的数据类型。

  • format – 指定矩阵格式的字符串（例如 ['bcoo']）。

  • **kwds – 传递给特定于格式的 _empty 构造函数的附加关键字。

  • sparse_format (str)

返回:

空的稀疏矩阵。

返回类型:

mat