Build the Laplacian matrix of a nearest neighbor graph.
tf_agents.bandits.agents.utils.build_laplacian_nearest_neighbor_graph(
input_vecs: tf_agents.typing.types.Tensor
,
k: int = 1
) -> tf_agents.typing.types.Tensor
Given input embedding vectors, this utility returns the Laplacian matrix of
the induced k-nearest-neighbor graph.
Args |
input_vecs
|
a Tensor . Input embedding vectors (one per row). Shaped
[num_vectors, ...] .
|
k
|
an integer. Number of nearest neighbors to use.
|
Returns |
The graph Laplacian matrix. A dense float Tensor of shape
[num_vectors, num_vectors] , where num_vectors is the number of input
embedding vectors (Tensor ).
|