Package io.github.kirstenali.deepj.activations

Interface ActivationFunction

All Known Implementing Classes:

GELU, ReLU, Sigmoid, Softmax, Tanh

public interface ActivationFunction

Method Summary

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

Method Details

forward

Tensor forward(Tensor input)

backward

Tensor backward(Tensor gradOutput)