flujo tensor:: operaciones:: Recolectar

#include <array_ops.h>

Reúna porciones 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 indices.shape + params.shape[1:] donde: 

    # Scalar indices
    output[:, ..., :] = params[indices, :, ... :]

    # Vector indices
    output[i, :, ..., :] = params[indices[i], :, ... :]

    # Higher rank indices
    output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]

Si indices es una permutación y len(indices) == params.shape[0] entonces esta operación permutará params en consecuencia.

validate_indices : DEPRECADO. Si esta operación se asigna a la CPU, los valores en indices siempre se validan para que estén dentro del rango. Si se asignan a GPU, los índices fuera de límites dan como resultado un comportamiento seguro pero no especificado, que puede incluir generar un error.

Argumentos:

Devoluciones:

  • Output : El tensor de salida.

Constructores y destructores

Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices)
Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, const Gather::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

ValidateIndices (bool x)

estructuras

tensorflow:: operaciones:: Reunir:: Atributos

Configuradores de atributos opcionales para Gather .

Atributos públicos

operación 

Operation operation

producción 

::tensorflow::Output output

Funciones públicas

Recolectar 

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

Recolectar 

 Gather(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  const Gather::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

Validar índices 

Attrs ValidateIndices(
  bool x
)