tf.gather

View source

金铨达配资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:

output$$[p0, ..., p{axis-1}, \hspace{5.1em} p{axis + 1}, ..., p{N-1}]$$ =\ params$$[p0, ..., p{axis-1}, \hspace{1em} indices, \hspace{1em} p{axis + 1}, ..., p{N-1}]$$.

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

output$$[p0, ..., p{axis-1}, \hspace{2.6em} i, \hspace{2.6em} p{axis + 1}, ..., p{N-1}]$$ =\ params$$[p0, ..., p{axis-1}, \hspace{1em} indices[i], \hspace{1em} p{axis + 1}, ..., p{N-1}]$$.

金铨达配资In the general case, produces an output tensor where:

$$\begin{align} output[p0, &..., p{axis-1}, & &i{B}, ..., i{M-1}, & p{axis + 1}, &..., p{N-1}] = \ params[p0, &..., p{axis-1}, & indices[p0, ..., p{B-1}, &i{B}, ..., i{M-1}], & p{axis + 1}, &..., p{N-1}] \end{align}$$

Where $$N$$=ndims(params), $$M$$=ndims(indices), and $$B$$=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.

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 rank(indices).
  • name: A name for the operation (optional).

Returns:

A Tensor. Has the same type as params.

results matching ""

    No results matching ""