# Codeblock 3 Outputoriginal          : torch.Size([1, 512, 28, 28])    #(1)after avgpool     : torch.Size([1, 512, 1, 1])      #(2)after flatten     : torch.Size([1, 512])            #(3)after fc0-relu    : torch.Size([1, 32])             #(4)after fc1-sigmoid : torch.Size([1, 512])            #(5)after reshape     : torch.Size([1, 512, 1, 1])      #(6)after scaling     : torch.Size([1, 512, 28, 28])    #(7)