Selecciona los k centros más cercanos para cada punto.
Se supone que las filas de puntos son puntos de entrada. Se supone que las filas de centros son la lista de centros candidatos. Para cada punto, se calculan los k centros que tienen la menor distancia L2 hacia él.
Constantes
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
estáticas NearestNeighbors | |
Salida < TFloat32 > | nearestCenterDistances () Matriz de forma (n, min (m, k)). |
Salida < TInt64 > | nearestCenterIndices () Matriz de forma (n, min (m, k)). |
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
Métodos públicos
public static NearestNeighbors crean ( Alcance alcance, operando < TFloat32 > puntos, operando < TFloat32 > centros, operando < TInt64 > k)
Método de fábrica para crear una clase que envuelva una nueva operación Vecinos más cercanos.
Parámetros
alcance | alcance actual |
---|---|
puntos | Matriz de forma (n, d). Se supone que las filas son puntos de entrada. |
centros | Matriz de forma (m, d). Se supone que las filas son centros. |
k | Número de centros más cercanos a devolver por cada punto. Si k es mayor que m, solo se devuelven m centros. |
Devoluciones
- una nueva instancia de NeighborsNeighbors
pública de salida < TFloat32 > nearestCenterDistances ()
Matriz de forma (n, min (m, k)). Cada fila contiene la distancia L2 al cuadrado del centro correspondiente en los índices del centro más cercano.