attention_layer.py 6.54 KB