Function marian::argmax

• Defined in File expression_operators.cpp

Function Documentation

Expr2 marian::argmax(Expr a, int axis)

Returns largest elements of an expression along an axis.

Return a 2-tuple (values, indices) of largest elements of an expression along a specified axis.

See

topk(a, k=1, axis, descending=true)