# Codeblock 4 Outputoriginal      : torch.Size([1, 3, 384, 384])after conv    : torch.Size([1, 768, 24, 24])  #(1)after flatten : torch.Size([1, 768, 576])     #(2)after permute : torch.Size([1, 576, 768])     #(3)