class GradReverse(Function):
def forward(self, x):
return x.view_as(x)
def backward(self, grad_output):
    return (grad_output * -lambd)

def grad_reverse(x):
return GradReverse()(x)
running the above snippet of code gives the error
class GradReverse(Function):
TypeError: Error when calling the metaclass bases
module.init() takes at most 2 arguments (3 given)
but i gave single 2d tensor as input,
what are the two inputs it takes,   x,lambd ?