TopKWithUnique

TopKWithUnique public final class

Devuelve los valores de TopK en la matriz en orden. Esta es una combinación

de MakeUnique y TopKUnique. El top-K devuelto tendrá sus bits inferiores reemplazados por iota, por lo que estará cerca del valor original pero no exactamente igual. El tiempo de ejecución es proporcional al producto de K por el tamaño de entrada. Los NaN nunca se devuelven. Los números subnormales se vacían a cero.

Constantes

Cuerda OP_NAME El nombre de esta operación, como lo conoce el motor central de TensorFlow

Métodos públicos

estática TopKWithUnique
crear ( Alcance alcance, operando < TFloat32 > entrada, Long k)
Método de fábrica para crear una clase que envuelva una nueva operación TopKWithUnique.
Salida < TFloat32 >
TopK ()
Salida < TInt32 >

Métodos heredados

Constantes

OP_NAME pública final static String

El nombre de esta operación, como lo conoce el motor central de TensorFlow

Valor constante: "TopKWithUnique"

Métodos públicos

public static TopKWithUnique crear ( Alcance alcance, operando < TFloat32 > Entrada, Long k)

Método de fábrica para crear una clase que envuelva una nueva operación TopKWithUnique.

Parámetros
alcance alcance actual
Devoluciones
  • una nueva instancia de TopKWithUnique

pública de salida < TFloat32 > TopK ()

pública de salida < TInt32 > topkIndices ()