Gather slices from params axis axis according to indices.
Aliases:
tf.compat.v2.gather
tf.gather(
params,
indices,
validate_indices=None,
axis=None,
batch_dims=0,
name=None
)
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:
For 1-D (vector) indices with batch_dims=0:
In the general case, produces an output tensor where:
Where
The shape of the output tensor is:
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.
tf.gather_ndSee also .
Args:
params: TheTensorfrom which to gather values. Must be at least rankaxis + 1.
Returns:
A Tensor. Has the same type as params.