mlx.nn.HardShrink

目录

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