# Codeblock 4 Output
original                : torch.Size([1, 512, 28, 28])

x after maxpool (x_max) : torch.Size([1, 512, 1, 1])
x after avgpool (x_avg) : torch.Size([1, 512, 1, 1])

x_max after flatten     : torch.Size([1, 512])
x_avg after flatten     : torch.Size([1, 512])

x_max after mlp         : torch.Size([1, 512])    #(1)
x_avg after mlp         : torch.Size([1, 512])    #(2)

after sum               : torch.Size([1, 512])
after sigmoid           : torch.Size([1, 512])
after reshape           : torch.Size([1, 512, 1, 1])    #(3)
after multiply          : torch.Size([1, 512, 28, 28])  #(4)