1import torch 2 3 4avg_pool1 = torch.nn.AdaptiveAvgPool2d((1, None)) 5reveal_type(avg_pool1) # E: {AdaptiveAvgPool2d} 6avg_pool2 = torch.nn.AdaptiveAvgPool2d((None, 1)) 7reveal_type(avg_pool2) # E: {AdaptiveAvgPool2d} 8max_pool1 = torch.nn.AdaptiveMaxPool2d((1, None)) 9reveal_type(max_pool1) # E: {AdaptiveMaxPool2d} 10max_pool2 = torch.nn.AdaptiveMaxPool2d((None, 1)) 11reveal_type(max_pool2) # E: {AdaptiveMaxPool2d} 12