Keras是否支持einsum？

Question

我在Keras尝试了tf.einsum（）：

t1 = Input(...)
t2 = Input(...)
t3 = tf.einsum('ijk,ij->ijk',t1, t2)
model = Model(input=[t1, t2], output=t3)

这会引发错误，＆＃39;属性错误：＆＃39; Tensor＆＃39;对象没有属性＆＃39; _keras_history＆＃39;。我发现t3没有_keras_history并且怀疑Keras不支持tf.einsum（）。

在Keras，有没有相当于einsum的东西？

Answer 1

keras模型中的每个操作都必须在keras层内完成：

t3 = Lambda(lambda x: tf.einsum('ijk,ij->ijk',x[0], x[1]))([t1,t2])