vision_transformer.py 9.7 KB