layer_sparsemax: Sparsemax activation function

In henry090/tfaddons: Interface to 'TensorFlow SIG Addons'

Sparsemax activation function

Description

Sparsemax activation function

Usage

layer_sparsemax(object, axis = -1, ...)

Arguments

object	Model or layer object
axis	Integer, axis along which the sparsemax normalization is applied.
...	additional parameters to pass

Details

The output shape is the same as the input shape. https://arxiv.org/abs/1602.02068

Value

A tensor

Examples

## Not run: 
model = keras_model_sequential() %>%
  layer_conv_2d(filters = 10, kernel_size = c(3,3),input_shape = c(28,28,1),
                activation = activation_gelu) %>%
  layer_sparsemax()

## End(Not run)

henry090/tfaddons documentation built on April 7, 2022, 11:27 p.m.