transformer_s2_layer.py 27.1 KB