attention_layer.py 5.17 KB