The tf.gather function allows you to gather slices from a tensor along a specified axis according to the indices provided. The
validate_indices argument is deprecated and setting its value has no effect. Indices are always validated on CPU and never validated on
GPU.