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

DecoderOutputs

DecoderOutputs​

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

source

Bases: GraphGroup[str, Logits | HiddenStates | ConvStates | SparseIndices]

A decoder graph’s named outputs: its logits, any hidden_states, and any state it writes back.

Parameters:

Return type:

Self

group()​

group(name, cls)

source

Returns the group named name, checked to be a cls.

Parameters:

Return type:

_G

hidden_states​

property hidden_states: HiddenStates

source

The decoder’s hidden states.

kind​

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

source

Identifies the feature wherever the group sits in a tree.

logits​

property logits: Logits

source

The decoder’s logits.