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
ReturnHiddenStates
ReturnHiddenStates
class max.nn.ReturnHiddenStates(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)
ALL
ALL = 'all'
ALL_NORMALIZED
ALL_NORMALIZED = 'all_normalized'
LAST
LAST = 'last'
LAST_PER_DEVICE
LAST_PER_DEVICE = 'last_per_device'
NONE
NONE = 'none'
SELECTED_LAYERS
SELECTED_LAYERS = 'selected_layers'