1 h1_c=h1[:,-1:,:] 2 tmp=tf.keras.backend.batch_dot(h1_c,tf.keras.layers.Permute((2,1))(h1)) 3 scores=tf.keras.layers.Softmax()(tmp) 4 at=tf.keras.backend.batch_dot(scores,h1)
tensorflow2.0、keras實現Attention
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。
猜您在找
基於tensorflow2.0 使用tf.keras實現Fashion MNIST
tensorflow2.0第2章 Tensorflow keras實戰
TensorFlow2.0教程1：keras 函數api
tensorflow2.0中引入keras和原來的keras的差異
tensorflow2.0學習——tensorflow-keras 歸一化
TensorFlow2.0教程-使用keras訓練模型
TensorFlow2.0教程2：使用keras訓練模型
TensorFlow2.0教程-使用keras訓練模型
TensorFlow2.0教程3：用keras構建自己的網絡層
TensorFlow2.0教程4：keras模型保存和序列化