Autoencoder types

Autoencoder types.

ResetOptimizerParameterDetails

Bases: NamedTuple

Reset Optimizer Parameter Details.

Details of a parameter that should be reset in the optimizer, when resetting its corresponding dictionary vectors.

Source code in sparse_autoencoder/autoencoder/types.py

class ResetOptimizerParameterDetails(NamedTuple):
    """Reset Optimizer Parameter Details.

    Details of a parameter that should be reset in the optimizer, when resetting
    its corresponding dictionary vectors.
    """

    parameter: Parameter
    """Parameter to reset."""

    axis: int
    """Axis of the parameter to reset."""

axis: int instance-attribute

Axis of the parameter to reset.

parameter: Parameter instance-attribute

Parameter to reset.