mlx.nn.Softshrink

Contents

mlx.nn.Softshrink#

class Softshrink(lambd=0.5)#

Applies the Softshrink function.

See softshrink() for the functional equivalent.

Parameters:

lambd – the \(\lambda\) value for Softshrink. Default: 0.5

Methods