tensorflow2.0、keras实现Attention

本文转载自查看原文 2020-11-03 09:22 600 随笔

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)

免责声明！

本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱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模型保存和序列化