mlx.nn.HardShrink#
- class HardShrink#
Applies the HardShrink function.
See
hard_shrink()for the functional equivalent.- 参数:
lambd -- the \(\lambda\) value for Hardshrink. Default:
0.5
Methods
Applies the HardShrink function.
See hard_shrink() for the functional equivalent.
lambd -- the \(\lambda\) value for Hardshrink. Default: 0.5
Methods