attention icon indicating copy to clipboard operation
attention copied to clipboard

Published 20 hours ago • verified publisher icon bojone

Keras的batch_dot()如何用pytorch改写

Open linjingxu opened this issue 4 years ago • 2 comments

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

linjingxu avatar Apr 25 '20 14:04 linjingxu

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

linjingxu avatar Apr 25 '20 16:04 linjingxu

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

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

Kuzma1998 avatar May 24 '21 12:05 Kuzma1998