tf.gather

Gather slices from params axis axis according to indices.

tf.gather(
    params, indices, validate_indices=None, name=None, axis=None, batch_dims=0
)

Gather slices from params axis axis according to indices. indices must be an integer tensor of any dimension (usually 0-D or 1-D).

For 0-D (scalar) indices:

output

=\

params

.

For 1-D (vector) indices with batch_dims=0:

output

=\

params

.

In the general case, produces an output tensor where:

Where

=ndims(params),

=ndims(indices), and

=batch_dims. Note that params.shape[:batch_dims] must be identical to indices.shape[:batch_dims].

The shape of the output tensor is:

output.shape = params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:].

Note that on CPU, if an out of bound index is found, an error is returned. On GPU, if an out of bound index is found, a 0 is stored in the corresponding output value.

See also tf.gather_nd.