Skip to main content

Python class

ReturnLogits

ReturnLogits​

class max.nn.ReturnLogits(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

source

Bases: str, Enum

ALL​

ALL = 'all'

source

LAST_TOKEN​

LAST_TOKEN = 'last_token'

source

VARIABLE​

VARIABLE = 'variable'

source