Fork me on GitHub

src/arraymancer/nn/activation/softmax

Theme:

Index

Search:

Group by:

Imports
Types
- SoftmaxActivation
Procs

Source Edit

Imports

autograd, tensor, nn_primitives

Types

SoftmaxActivation[TT] = ref object of Gate[TT]: Source Edit

Procs

proc softmax[TT](a: Variable[TT]): Variable[TT]: Input:
A variable

Source Edit

Arraymancer Technical reference

Core tensor API

Neural network API

Linear algebra, stats, ML

IO & Datasets

Autograd

Neuralnet primitives

Other docs

Tutorial

Spellbook (How-To's)

Under the hood