Learn R Programming

reservr (version 0.0.2)

softmax: Soft-Max function

Description

Softmax for a vector x is defined as

Usage

softmax(x)

dsoftmax(x)

Value

softmax returns the softmax of x; rowwise if x is a matrix.

dsoftmax returns the Jacobi-matrix of softmax(x) at x. x must be a vector.

Arguments

x

A numeric vector or matrix

Details

\(s_i = \exp(x_i) / \sum_k \exp(x_k)\)

It satisfies sum(s) == 1.0 and can be used to smoothly enforce a sum constraint.

Examples

Run this code
softmax(c(5, 5))
softmax(diag(nrow = 5, ncol = 6))

Run the code above in your browser using DataLab