flujo tensor:: operaciones:: Recolectar

#include <array_ops.h>

Reúna porciones de params según 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
)