Keras中 用Lambda层封装的矩阵乘法
Keras中 用Lambda层封装的矩阵乘法keras中对于层的封装绝对是一个大坑,今天分享一些自己在写代码中遇到的bug。如果是对层进行矩阵乘法,假设输入层分别为A:(batch_size, 300, 64),B:(batch_size,300,64),我们需要对A进行转置再与B进行矩阵乘法,但是Keras中无法直接对tensor进行矩阵乘法,会报错,必须要对其进行封装。# A:(batch_size, 300, 64),B:(batch_size,300,64)A = Permute((2, 1)
原创
2021-01-05 19:53:24 ·
2079 阅读 ·
2 评论