softmax: softmax

Description Usage Arguments Examples

Description

uses logsumexp trick to prevent numerical overflow

Usage

1

Arguments

x

a vector of numbers

Examples

1
2
3
softmax2 <- function(x) exp(x) / sum(exp(x))
softmax(c(1, 2, 3) * 1000)  # NaN NaN NaN
softmax2(c(1, 2, 3) * 1000)  # 0 0 1

oslerinhealth/baker documentation built on May 22, 2021, 12:05 p.m.