Function marian::index_select(Expr, int, const std::vector<IndexType>&)

Function Documentation

Expr marian::index_select(Expr a, int axis, const std::vector<IndexType> &indices)

Returns a new expression containing the indicies of expression a along the specified axis. Convenience wrapper that promotes a vector of IndexType to an Expr