Working Of Vision Transformer Pytorch