# Codeblock 5 Outputoriginal            : torch.Size([1, 96, 56, 56])residual after proj : torch.Size([1, 192, 28, 28])  #(1)after conv0         : torch.Size([1, 192, 56, 56])  #(2)after permute       : torch.Size([1, 56, 56, 192])after norm0         : torch.Size([1, 56, 56, 192])after permute       : torch.Size([1, 192, 56, 56])after conv1         : torch.Size([1, 768, 56, 56])after gelu          : torch.Size([1, 768, 56, 56])after conv2         : torch.Size([1, 192, 56, 56])  #(3)after permute       : torch.Size([1, 56, 56, 192])after norm1         : torch.Size([1, 56, 56, 192])after permute       : torch.Size([1, 192, 56, 56])after downsample    : torch.Size([1, 192, 28, 28])  #(4)after summation     : torch.Size([1, 192, 28, 28])  #(5)