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

HiddenStates

`HiddenStates`

class max.nn.legacy.graph_groups.HiddenStates(*args, **kwargs)

source

Hidden states a decoder returns for one mode.

The mode is a drafter, a capture, or the next pipeline stage.

Creates the group, rejecting a subclass with no kind.

Parameters:

Return type:

Self

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

source

Identifies the feature wherever the group sits in a tree.

tensors()

source

Returns the hidden states, whichever mode produced them.