Keras - 如何计算双和？

Question

我正在使用keras，我必须实现像this one.

这样的双重和

其中f是（α，N_f）张量，g是（α，N_g）张量。乘积f * g应为（？，N_f * N_g）张量，减少的双和为a（1，）张量。

我如何在Keras上实现这一点？

谢谢

Answer 1

以下是我最终的结果：

def double_sum(x, y): 
    #n_x should be equal to n_y
    n_x = x.shape[1].value
    n_y = y.shape[1].value

    x_rep = K.repeat_elements(x, n_x, axis=1)
    y_tile = K.tile(y, [1, n_y])
    return K.sum(K.dot(x_rep, y_tile))

我希望将来帮助其他人。