Class LogSoftmax

java.lang.Object

smile.deep.activation.ActivationFunction

smile.deep.activation.LogSoftmax

All Implemented Interfaces:

Serializable, Function<Tensor,Tensor>, Layer

public class LogSoftmax
extends ActivationFunction

Log softmax activation function.

See Also:

• Serialized Form

Constructor Summary

Constructors

Constructor	Description
LogSoftmax()	Constructor.
LogSoftmax(int dim)	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

LogSoftmax

public LogSoftmax()

Constructor. Computes log-softmax along dimension 1 (class dimension for 2-D batch × classes tensors).

LogSoftmax

public LogSoftmax(int dim)

Constructor.

Parameters:

dim - the dimension along which log-softmax is computed.

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.