attention_layer.py 5.13 KB