Function marian::softmax(Expr, Expr, int)

• Defined in File expression_operators.cpp

Function Documentation

Expr marian::softmax(Expr a, Expr zeroOneMask, int axis = -1)

Computes the softmax fuction along the given axis. Applies the softmax function over the unmasked values.

See

SoftmaxNodeOp