tensorflow:: operaciones:: Recolectar

#include <array_ops.h>

Reúna rebanadas de params de acuerdo con los indices .

Resumen

los indices deben ser un tensor entero de cualquier dimensión (generalmente 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 : DESAPROBADO. Si esta operación se asigna a la CPU, los valores de los indices siempre se validan para estar dentro del rango. Si se asignan a GPU, los índices fuera de límite 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 publicas

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

Funciones estáticas públicas

ValidateIndices (bool x)

estructuras

tensorflow:: ops:: Gather:: Atributos

Establecedores de atributos opcionales para Gather .

Atributos públicos

operación

Operation operation

producción

::tensorflow::Output output

Funciones publicas

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

ValidateIndices

Attrs ValidateIndices(
  bool x
)