tensorflow :: operaciones :: TopK

#include <nn_ops.h>

Encuentra valores e índices de los k elementos más grandes para la última dimensión.

Resumen

Si la entrada es un vector (rango-1), encuentra las k entradas más grandes en el vector y genera sus valores e índices como vectores. Por lo tanto, los values[j] son la j -ésima entrada más grande en la input , y su índice son los indices[j] .

Para matrices (resp. Entrada de rango superior), calcula las k entradas superiores en cada fila (resp. Vector a lo largo de la última dimensión). Por lo tanto,

values.shape = indices.shape = input.shape[:-1] + [k]

Si dos elementos son iguales, el elemento de índice más bajo aparece primero.

Argumentos:

  • alcance: un objeto de alcance
  • entrada: 1-D o superior con la última dimensión al menos k .
  • k: 0-D. Número de elementos superiores a buscar a lo largo de la última dimensión (a lo largo de cada fila para matrices).

Atributos opcionales (consulte Attrs ):

  • sorted: si es verdadero, los k elementos resultantes serán ordenados por los valores en orden descendente.

Devoluciones:

  • Valores de Output : los k elementos más grandes a lo largo de cada último corte dimensional.
  • Índices de Output : los índices de values dentro de la última dimensión de input .

Constructores y Destructores

TopK (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input k)
TopK (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input k, const TopK::Attrs & attrs)

Atributos públicos

indices
operation
values

Funciones estáticas públicas

Sorted (bool x)

Estructuras

tensorflow :: ops :: TopK :: Attrs

Establecedores de atributos opcionales para TopK .

Atributos públicos

índices

::tensorflow::Output indices

operación

Operation operation

valores

::tensorflow::Output values

Funciones publicas

TopK

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k
)

TopK

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k,
  const TopK::Attrs & attrs
)

Funciones estáticas públicas

Ordenados

Attrs Sorted(
  bool x
)