def __init__(self, kernel_size=(3, 3), stride=2, padding=1, dim=128):
    super(Max_AvgPool, self).__init__()
    self.Maxpool = nn.MaxPool2d(kernel_size=kernel_size, stride=stride, padding=padding)
    self.Avgpool = nn.AvgPool2d(kernel_size=kernel_size, stride=stride, padding=padding)
def __init__(self, kernel_size=(3, 3), stride=2, padding=1, dim=128):
    super(Max_AvgPool, self).__init__()
    self.Maxpool = nn.MaxPool2d(kernel_size=kernel_size, stride=stride, padding=padding)
    self.Avgpool = nn.AvgPool2d(kernel_size=kernel_size, stride=stride, padding=padding)