nn_contrib_sparsemax: Sparsemax activation

nn_contrib_sparsemaxR Documentation

Sparsemax activation

Description

Sparsemax activation module.

Usage

nn_contrib_sparsemax(dim = -1)

Arguments

dim

The dimension over which to apply the sparsemax function. (-1)

Details

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


torch documentation built on May 29, 2024, 9:54 a.m.