Class TorchLinearActivation

numpy_ml/tests/nn_torch_models.py:124–135

class TorchLinearActivation(nn.Module):
    def __init__(self):
        super(TorchLinearActivation, self).__init__()
        pass

    @staticmethod
    def forward(input):
        return input

    @staticmethod
    def backward(grad_output):
        return torch.ones_like(grad_output)


class TorchBatchNormLayer(nn.Module):