I also have the same question. In my case, I use the code below. Can anyone help?
class Maxout(nn.Module):
         def __init__(self):
              super(Maxout, self).__init__()
        def forward(self, x, y):
              return torch.max(x,y)```