如何解决MultiheadAttention 的输入维度在 Pytorch 和 Tensorflow “语言理解的变压器模型”之间有所不同
Pytorch 实现的 MultiheadAttention,TransformerEncoder,Batch size 居中。即查询维度的 (L,N,E)。 但是在Tensorflow的《语言理解的Transformer Model》一文中,查询维度是(batch_size,seq_len,depth)。该 batch_size 位于第 0 个索引处。 为什么会有所不同?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。