tensorflow::ops::Gather

tensorflow::ops::Gather

#include <array_ops.h>

Gather slices from params according to indices.

Summary

indices must be an integer tensor of any dimension (usually 0-D or 1-D). Produces an output tensor with shape indices.shape + params.shape[1:] where:

```python Scalar indices

output[:, ..., :] = params[indices, :, ... :]

Vector indices

output[i, :, ..., :] = params[indices[i], :, ... :]

Higher rank indices

output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :] ```

If indices is a permutation and len(indices) == params.shape[0] then this operation will permute params accordingly.

validate_indices: DEPRECATED. If this operation is assigned to CP