Method __init__

numpy_ml/tests/nn_torch_models.py:302–304

(self, act_fn, **kwargs)

class TorchAddLayer(nn.Module):
    def __init__(self, act_fn, **kwargs):
        super(TorchAddLayer, self).__init__()
        self.act_fn = act_fn

    def forward(self, Xs):
        self.Xs = []