文章目录 tensorflow版本 tensorflow实现 附录,torch实现 tensorflow版本 Tensorflow2.16 + python3.12 tensorflow实现 import tensorflow as tf class Attention(tf.keras.Layer): def __init__(self, in_heads