layer_sparsemax: Sparsemax activation function

In tfaddons: Interface to 'TensorFlow SIG Addons'

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)

tfaddons documentation built on July 2, 2020, 2:12 a.m.