org.tensorflow.op.image

Class NearestNeighbors

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.image.NearestNeighbors

All Implemented Interfaces:

Op

public final class NearestNeighbors
extends RawOp

Selects the k nearest centers for each point. Rows of points are assumed to be input points. Rows of centers are assumed to be the list of candidate centers. For each point, the k centers that have least L2 distance to it are computed.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	NearestNeighbors.Inputs

Field Summary

Fields

Modifier and Type	Field and Description
static String	OP_NAME The name of this op, as known by TensorFlow core engine

Fields inherited from class org.tensorflow.op.RawOp

operation

Constructor Summary

Constructors

Constructor and Description
NearestNeighbors(Operation operation)

Method Summary

Modifier and Type	Method and Description
static NearestNeighbors	create(Scope scope, Operand<TFloat32> points, Operand<TFloat32> centers, Operand<TInt64> k) Factory method to create a class wrapping a new NearestNeighbors operation.
Output<TFloat32>	nearestCenterDistances() Gets nearestCenterDistances.
Output<TInt64>	nearestCenterIndices() Gets nearestCenterIndices.

Methods inherited from class org.tensorflow.op.RawOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.tensorflow.op.Op

env

Field Detail

OP_NAME

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

See Also:

Constant Field Values

Constructor Detail

NearestNeighbors

public NearestNeighbors(Operation operation)

Method Detail

create

@Endpoint(describeByClass=true)
public static NearestNeighbors create(Scope scope,
                                                                      Operand<TFloat32> points,
                                                                      Operand<TFloat32> centers,
                                                                      Operand<TInt64> k)

Factory method to create a class wrapping a new NearestNeighbors operation.

Parameters:

scope - current scope

points - Matrix of shape (n, d). Rows are assumed to be input points.

centers - Matrix of shape (m, d). Rows are assumed to be centers.

k - Number of nearest centers to return for each point. If k is larger than m, then only m centers are returned.

Returns:

a new instance of NearestNeighbors

nearestCenterIndices

public Output<TInt64> nearestCenterIndices()

Gets nearestCenterIndices. Matrix of shape (n, min(m, k)). Each row contains the indices of the centers closest to the corresponding point, ordered by increasing distance.

Returns:

nearestCenterIndices.

nearestCenterDistances

public Output<TFloat32> nearestCenterDistances()

Gets nearestCenterDistances. Matrix of shape (n, min(m, k)). Each row contains the squared L2 distance to the corresponding center in nearest_center_indices.

Returns:

nearestCenterDistances.