View source: R/nnf-activation.R
nnf_contrib_sparsemax | R Documentation |
Applies the SparseMax activation.
nnf_contrib_sparsemax(input, dim = -1)
input |
the input tensor |
dim |
The dimension over which to apply the sparsemax function. (-1) |
The SparseMax activation is described in 'From Softmax to Sparsemax: A Sparse Model of Attention and Multi-Label Classification' The implementation is based on aced125/sparsemax
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.