Package smile.deep.activation

Class Softmax

java.lang.Object

smile.deep.activation.ActivationFunction

smile.deep.activation.Softmax

All Implemented Interfaces:

Serializable, Function<Tensor,Tensor>, Layer

public class Softmax
extends ActivationFunction

Softmax activation function.

See Also:

• Serialized Form

Constructor Summary

Constructors

Constructor	Description
Softmax()	Constructor.

Method Summary

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

Methods inherited from class smile.deep.activation.ActivationFunction

asTorch, isInplace, name

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.util.function.Function

andThen, compose

Methods inherited from interface smile.deep.layer.Layer

apply, isTraining

Constructor Details

Softmax

public Softmax()

Constructor.

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.