Uses of Interface
io.github.kirstenali.deepj.activations.ActivationFunction
Packages that use ActivationFunction
Package
Description
-
Uses of ActivationFunction in io.github.kirstenali.deepj.activations
Classes in io.github.kirstenali.deepj.activations that implement ActivationFunctionModifier and TypeClassDescriptionfinal classGaussian Error Linear Unit (GELU), using the tanh approximation popularized by GPT-2.classclassfinal classSigmoid Linear Unit (SiLU / Swish).final classRow-wise softmax for 2D tensors: applies softmax independently to each row.class -
Uses of ActivationFunction in io.github.kirstenali.deepj.layers
Constructors in io.github.kirstenali.deepj.layers with parameters of type ActivationFunctionModifierConstructorDescriptionFNN(int inputSize, int[] hiddenSizes, int outputSize, Supplier<ActivationFunction> hiddenActivationFactory, ActivationFunction outputActivation, Random rnd) Build an MLP of the form: Linear -> act -> Linear -> act -> ... -> Linear -> (optional outputAct)Constructor parameters in io.github.kirstenali.deepj.layers with type arguments of type ActivationFunctionModifierConstructorDescriptionFNN(int inputSize, int[] hiddenSizes, int outputSize, Supplier<ActivationFunction> hiddenActivationFactory, ActivationFunction outputActivation, Random rnd) Build an MLP of the form: Linear -> act -> Linear -> act -> ... -> Linear -> (optional outputAct)FNN(int inputSize, int[] hiddenSizes, int outputSize, Supplier<ActivationFunction> hiddenActivationFactory, Random rnd) -
Uses of ActivationFunction in io.github.kirstenali.deepj.layers.transformer.blocks
Constructor parameters in io.github.kirstenali.deepj.layers.transformer.blocks with type arguments of type ActivationFunctionModifierConstructorDescriptionGPTTransformerBlock(int dModel, int nHeads, int dFF, Supplier<ActivationFunction> ffnActivationFactory, Random rnd) Full constructor. -
Uses of ActivationFunction in io.github.kirstenali.deepj.transformer
Method parameters in io.github.kirstenali.deepj.transformer with type arguments of type ActivationFunctionModifier and TypeMethodDescriptionGPTTransformerBuilder.ffnActivation(Supplier<ActivationFunction> factory) Activation inside the FFN.