Function marian::gather

• Defined in File expression_operators.cpp

Function Documentation

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

Gathers elements along an axis.

Return

Gathered expression with the same shape as indices

Note

a and indices must have the same rank

Note

The non-target axes of a and indices must have the same size, or be broadcastable.

Parameters

• a: The input expression

• axis: The axis along which to index

• indices: The indices to be gathered