flujo tensor:: 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, values[j] es la j -ésima entrada más grande en input y su índice es indices[j] .

Para matrices (respectivamente, entrada de rango superior), calcula las k entradas principales en cada fila (respectivamente, 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 menor índice 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 (ver Attrs ):

  • ordenado: si es verdadero, los k elementos resultantes se ordenarán por valores en orden descendente.

Devoluciones:

  • Valores Output : los k elementos más grandes a lo largo de cada último segmento dimensional.
  • Índices 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:: operaciones:: TopK:: Atributos

Configuradores de atributos opcionales para TopK .

Atributos públicos

índices

::tensorflow::Output indices

operación

Operation operation

valores

::tensorflow::Output values

Funciones públicas

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

Ordenado

Attrs Sorted(
  bool x
)