attention_layer.py 6.95 KB