IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Python class

SparseIndices

SparseIndices​

class max.nn.graph_groups.SparseIndices(*args, **kwargs)

source

Bases: GraphGroup[str, tuple[TensorValue, …]]

The rows a sparse-attention indexer selected, for a drafter to reuse.

Parameters:

Return type:

Self

kind​

kind: ClassVar[str | None] = 'sparse_indices'

source

Identifies the feature wherever the group sits in a tree.