Package io.github.kirstenali.deepj.activations

Class Softmax

java.lang.Object

io.github.kirstenali.deepj.activations.Softmax

All Implemented Interfaces:

ActivationFunction

public final class Softmax
extends Object
implements ActivationFunction

Row-wise softmax for 2D tensors: applies softmax independently to each row. Forward: softmaxRows(logits) Backward: given upstream grad dY, returns dLogits using cached softmax output.

Constructor Summary

Constructors

Constructor	Description
Softmax()

Method Summary

Modifier and Type	Method	Description
Tensor	backward(Tensor gradOutput)
Tensor	forward(Tensor logits)

Methods inherited from class java.lang.Object

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

Constructor Details

Softmax

public Softmax()

Method Details

forward

public Tensor forward(Tensor logits)

Specified by:

forward in interface ActivationFunction

backward

public Tensor backward(Tensor gradOutput)

Specified by:

backward in interface ActivationFunction