Args:

params: The Tensor from which to gather values. Must be at least rank
axis + 1.

indices: The index Tensor. Must be one of the following types: int32,
int64. Must be in range [0, params.shape[axis]).

validate_indices: Deprecated, does nothing.

axis: A Tensor. Must be one of the following types: int32, int64. The
axis in params to gather indices from. Must be greater than or equal
to batch_dims. Defaults to the first non-batch dimension. Supports
negative indexes.

batch_dims: An integer. The number of batch dimensions. Must be less
than or equal to rank(indices).