attention_layer.py 5.22 KB