Se usó la API de Cloud Translation para traducir esta página.
Switch to English

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ón axis + 1 .
  • índices: tensor de índice. Debe estar en el rango [0, params.shape[axis]) .
  • eje: El eje en los params que se obtienen los indices . Por defecto es la primera dimensión. Admite índices negativos.

Devoluciones:

  • Output : valores de los params recopilados a partir de índices dados por indices , con forma params.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

salida

::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
)