自定义图层矩阵大小不兼容

Question

我正在测试自己的自定义层。这是噪声层的实现。但是，我不断收到错误：矩阵大小不兼容 我不知道是什么原因导致无法执行计算。

例如：

login

提高：

//App.vue

created() {
function login(email, password, reportID = null) {
    this.$http
        .post('login/', { email: email, password: password, reportID: reportID })
        .then(() =>{...})
    login('dummy@dummy.com', 'dummy', 2);
}

这是我的自定义图层的代码：

from  NoisyLayer import NoisyLayer
from keras.models import Model
from keras.layers import Input

inp = Input(shape=[100])
out = NoisyLayer(units=10)(inp)

model = Model(inp, out)
a = np.expand_dims(a, 0)
model.predict(a)

预期结果是： 其中weight_mu，weight_sigma，bias_mu和bias_sigma是图层的可训练参数，eps_w，eps_b是一些随机的不可训练参数，计算公式为： （@表示逐元素乘法）

InvalidArgumentError: Matrix size-incompatible: In[0]: [10,100], In[1]: [1,100]
 [[{{node noisy_layer_32/MatMul}}]]

输出= relu（y）