Keras的batch_dot()如何用pytorch改写

[linjingxu]

你好，我想把你的Keras版本改写成pytorch版本的，但是K.batch_dot(qw, kw, [4, 4])这个功能好像没有类似pytorch的函数可以转换，请问你有什么建议吗？谢谢

[linjingxu]

我已经解决啦，torch.matmul()也是可以对高维张量进行操作

[Kuzma1998]

我已经解决啦，torch.matmul()也是可以对高维张量进行操作

你好，请问可以发一份您的pytorch版本的学习一下吗，万分感谢!