Keras - 如何计算双倍和？

Question

我正在使用 keras，我必须实现像 this one. 这样的双倍总和

其中 f 是 (?, N_f) 张量，g 是 (?, N_g) 张量。乘积 f*g 应该是 (?, N_f * N_g) 张量，而约化的双和应该是 (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))

希望以后能帮到别人。