stable-log-softmax ( seq -- softmax )

Vocabulary

machine-learning.functions

Inputs

seq

an object

Outputs

softmax

an object

Definition

USING: kernel math.functions math.vectors sequences ;

IN: machine-learning.functions

: stable-log-softmax ( seq -- softmax )
    dup maximum v-n dup [ e^ ] map-sum log v-n ;