flujo tensor:: operaciones:: TopK

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). De este modo,

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 .

Atributos públicos

::tensorflow::Output indices

Operation operation

::tensorflow::Output values

 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
)

Attrs Sorted(
  bool x
)