flujo tensor:: operaciones:: ReunirV2

#include <array_ops.h>

Reúna cortes del axis de params según los indices .

Resumen

indices deben ser un tensor entero de cualquier dimensión (normalmente 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 de límite, se devuelve un error. En GPU, si se encuentra un índice fuera de 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 cual recolectar valores. Debe tener al menos un rango de axis + 1 .
  • índices: tensor de índice. Debe estar en el rango [0, params.shape[axis]) .
  • eje: el eje en params del que recopilar indices . El valor predeterminado es la primera dimensión. Soporta índices negativos.

Devoluciones:

  • Output : valores de params recopilados 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 públicas

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Funciones estáticas públicas

BatchDims (int64 x)

estructuras

tensorflow:: operaciones:: GatherV2:: Atributos

Configuradores de atributos opcionales para GatherV2 .

Atributos públicos

operación 

Operation operation

producción 

::tensorflow::Output output

Funciones públicas

ReunirV2 

 GatherV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  ::tensorflow::Input axis
)

ReunirV2 

 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

Atenuaciones por lotes 

Attrs BatchDims(
  int64 x
)