Compute the nearest neighbor of the input queries. Arguments: x: A normalized matrix of queries of size (batch_size x key_dim) y: A matrix of correct labels (batch_size x 1) Returns: y_hat, A (batch-size x 1) matrix - the neare
(self, x, y, predict=False)
source not stored for this graph (policy: none)
no test coverage detected