Function marian::logsumexp

Function Documentation

Expr marian::logsumexp(Expr a, int ax)

Compute the log of the sum of exponentials along the specified axis.

See

ReduceNodeOp

Parameters
  • ax: Axis along which to perform the operation