Class SoftShrink

java.lang.Object

smile.deep.activation.ActivationFunction

smile.deep.activation.SoftShrink

All Implemented Interfaces:

Serializable, Function<Tensor,Tensor>, Layer

public class SoftShrink
extends ActivationFunction

Soft Shrink activation function.

See Also:

• Serialized Form

Constructor Summary

Constructors

Constructor	Description
SoftShrink()	Constructor.
SoftShrink(double lambda)	Constructor.

Method Summary

Modifier and Type	Method	Description
Tensor	forward(Tensor x)	Forward propagation (or forward pass) through the layer.

Methods inherited from class ActivationFunction

asTorch, isInplace, name

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface Function

andThen, compose

Methods inherited from interface Layer

apply, to, to

Constructor Details

SoftShrink

public SoftShrink()

Constructor.

SoftShrink

public SoftShrink(double lambda)

Constructor.

Parameters:

lambda - The lambda value in the formulation.

Method Details

forward

public Tensor forward(Tensor x)

Description copied from interface: Layer

Forward propagation (or forward pass) through the layer.

Parameters:

x - the input tensor.

Returns:

the output tensor.