tensorflow:: ops:: GatherV2
#include <array_ops.h>
Gather slices from params
axis axis
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 params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]
where:
# Scalar indices (output is rank(params) - 1). output[a_0, ..., a_n, b_0, ..., b_n] = params[a_0, ..., a_n, indices, b_0, ..., b_n]
# Vector indices (output is rank(params)). output[a_0, ..., a_n, i, b_0, ..., b_n] = params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
# Higher rank indices (output is rank(params) + rank(indices) - 1). output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] = params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]
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.batch_gather
and tf.gather_nd
.
Args:
- scope: A Scope object
- params: The tensor from which to gather values. Must be at least rank
axis + 1
. - indices: Index tensor. Must be in range
[0, params.shape[axis])
. - axis: The axis in
params
to gatherindices
from. Defaults to the first dimension. Supports negative indexes.
Returns:
Output
: Values fromparams
gathered from indices given byindices
, with shapeparams.shape[:axis] + indices.shape + params.shape[axis + 1:]
.
Constructors and Destructors |
|
---|---|
GatherV2(const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices, ::tensorflow::Input axis)
|
|
GatherV2(const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices, ::tensorflow::Input axis, const GatherV2::Attrs & attrs)
|
Public attributes |
|
---|---|
operation
|
|
output
|
Public functions |
|
---|---|
node() const
|
::tensorflow::Node *
|
operator::tensorflow::Input() const
|
|
operator::tensorflow::Output() const
|
|
Public static functions |
|
---|---|
BatchDims(int64 x)
|
Structs |
|
---|---|
tensorflow:: |
Optional attribute setters for GatherV2. |
Public attributes
operation
Operation operation
output
::tensorflow::Output output
Public functions
GatherV2
GatherV2( const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices, ::tensorflow::Input axis )
GatherV2
GatherV2( const ::tensorflow::Scope & scope, ::tensorflow::Input params, ::tensorflow::Input indices, ::tensorflow::Input axis, const GatherV2::Attrs & attrs )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
Public static functions
BatchDims
Attrs BatchDims( int64 x )