optax.scale#
- optax.scale(
- step_size: jax.typing.ArrayLike,
Scale updates by some fixed scalar step_size.
- Parameters:
step_size – A scalar corresponding to a fixed scaling factor for updates.
- Returns:
A
optax.GradientTransformationobject.
Scale updates by some fixed scalar step_size.
step_size – A scalar corresponding to a fixed scaling factor for updates.
A optax.GradientTransformation object.