GRU单元门的权重维数是多少?

时间:2018-01-13 15:51:12

标签: python deep-learning lasagne

我正在Lasagne中实现一个神经网络,我想在不同的GRU层(http://lasagne.readthedocs.io/en/latest/modules/layers/recurrent.html#lasagne.layers.GRULayer)之间共享权重。为此,我通过自定义门(http://lasagne.readthedocs.io/en/latest/modules/layers/recurrent.html#lasagne.layers.Gate)替换GRU图层的重置,更新和隐藏更新门。

在这些门中,我必须定义输入到门的权重W_in和隐藏到门的权重W_hid。这些权重的维度应该是多少?我最好的猜测是,对于具有维度的输入数据,batch_size x input_len x num_features,dim(W_in)= num_features x num_hidden和dim(W_hid)= num_hidden x num_hidden。但是,这不起作用。

有人有想法吗?提前谢谢!

0 个答案:

没有答案