Lambda

class torchelie.nn.Lambda(lam)

Applies a lambda function on forward()

Parameters

lamb (fn) – the lambda function

forward(x)
training: bool