Lambda

class torchelie.nn.Lambda(lam: torch.nn.modules.module.Module)

Applies a lambda function on forward()

Parameters

lamb (fn) – the lambda function

forward(x)

lam: torch.nn.modules.module.Module