Method __init__

beginner_source/nn_tutorial.py:743–745

(self, func)

class Lambda(nn.Module):
    def __init__(self, func):
        super().__init__()
        self.func = func

    def forward(self, x):
        return self.func(x)