An easy way to create a pytorch layer for a simple `func`

Lambda(func)

Arguments

func

function

Value

None