mlx.nn.Softshrink

目录

mlx.nn.Softshrink#

class Softshrink(lambd=0.5)#

Applies the Softshrink function.

See softshrink() for the functional equivalent.

参数:

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

Methods