attention_layer.py 6.8 KB