tensorflow :: operaciones :: GatherV2
#include <array_ops.h>
Reúna cortes del eje del axis
de params
acuerdo con los indices
.
Resumen
indices
deben ser un tensor entero de cualquier dimensión (generalmente 0-D o 1-D). Produce un tensor de salida con forma params.shape[:axis] + indices.shape + params.shape[axis + 1:]
donde:
# 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]
Tenga en cuenta que en la CPU, si se encuentra un índice fuera del límite, se devuelve un error. En la GPU, si se encuentra un índice fuera del límite, se almacena un 0 en el valor de salida correspondiente.
Consulte también tf.batch_gather
y tf.gather_nd
.
Argumentos:
- alcance: un objeto de alcance
- params: el tensor del que se obtienen los valores. Debe ser al menos el
axis + 1
clasificaciónaxis + 1
. - índices: tensor de índice. Debe estar en el rango
[0, params.shape[axis])
. - eje: El eje en los
params
que se recopilan losindices
. Por defecto es la primera dimensión. Admite índices negativos.
Devoluciones:
-
Output
: Valores de losparams
recopilados a partir de índices dados porindices
, con formaparams.shape[:axis] + indices.shape + params.shape[axis + 1:]
.
Constructores y Destructores | |
---|---|
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) |
Atributos públicos | |
---|---|
operation | |
output |
Funciones publicas | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Funciones estáticas públicas | |
---|---|
BatchDims (int64 x) |
Estructuras | |
---|---|
tensorflow :: ops :: GatherV2 :: Attrs | Establecedores de atributos opcionales para GatherV2 . |
Atributos públicos
operación
Operation operation
producción
::tensorflow::Output output
Funciones publicas
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 )
nodo
::tensorflow::Node * node() const
operador :: tensorflow :: Entrada
operator::tensorflow::Input() const
operador :: tensorflow :: Salida
operator::tensorflow::Output() const
Funciones estáticas públicas
BatchDims
Attrs BatchDims( int64 x )