k1lib.nn module¶

Some nice utils to complement torch.nn

class k1lib.nn.Lambda(f: Callable[[Any], Any])[source]¶

Bases: torch.nn.modules.module.Module

__init__(f: Callable[[Any], Any])[source]¶: Creates a simple module with a specified forward() function.

class k1lib.nn.Identity[source]¶

Bases: k1lib.nn.Lambda

Creates a module that returns the input in forward()