k1lib.knn module

Some nice utils to complement torch.knn

class k1lib.knn.Lambda(f: Callable[[Any], Any])

Bases: torch.nn.modules.module.Module

__init__(f: Callable[[Any], Any])

Creates a simple module with a specified forward() function.

class k1lib.knn.Identity

Bases: k1lib.knn.Lambda

Creates a module that returns the input in forward()