实现 ViT 的图像分块与 Patch Embedding。