attention_layer.py 6.27 KB