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

Function Documentation

Expr marian::index_select(Expr a, int axis, Expr indices)

Returns a new expression containing the indicies of expression a along the specified axis.

Warning

Do not pass a scalar literal 0 as indices; it will compile but pass a nullptr.