Keras поддерживает einsum?

Question

Я пробовал tf.einsum () в Керасе:

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

Это выдает ошибку: «AttributeError: объект Tensor» не имеет атрибута «_keras_history». Я обнаружил, что t3 не имеет _keras_history и подозреваю, что tf.einsum () не поддерживается в Keras.

В Керасе, есть что-нибудь, что эквивалентно einsum?

Answer 1

Каждая операция в модели keras должна выполняться внутри слоя keras:

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