tensorflow:: ops:: TopK

Encuentra los 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 (o entrada de rango superior), calcula las k entradas principales en cada fila (o 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:

• ámbito: Un objeto de ámbito
• Entrada: 1-D o superior con la última dimensión de 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 sus valores en orden descendente.

Devoluciones:

• Valores Output : Los k elementos más grandes a lo largo de cada segmento dimensional anterior.
• Í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
)