如何在批量数据上使用点生产?

时间:2016-07-12 06:15:11

标签: python numpy theano deep-learning keras

我正在尝试应用 tanh(dot(x,y)); x和y是我的RNN的批量数据。

x,y具有形状(n_batch,n_length,n_dim),如(2,3,4); 2个样本有3个序列,每个是4维。

我想做内部或点生产到最后的维度。然后tanh(dot(x,y))的形状应为(n_batch,n_length)=(2,3)

我应该使用哪种功能?

1 个答案:

答案 0 :(得分:1)

这个表达式可以解决这个问题:

  

theano.tensor.tanh((x * y).sum(2))

通过逐元素乘法“手动”计算点积,然后对最后一维进行求和。