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

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

after concatenate       : torch.Size([1, 2, 28, 28])    #(3)
after conv              : torch.Size([1, 1, 28, 28])    #(4)
after sigmoid           : torch.Size([1, 1, 28, 28])
after multiply          : torch.Size([1, 512, 28, 28])  #(5)