Keras, K.repeat_elements приводит к AttributeError: у объекта 'Tensor' нет атрибута '_keras_history'

Question

У меня есть слой с размером (None, 1, B), и я хотел бы клонировать его строки так, чтобы я получил что-то вроде (None, A, B). Моя идея заключается в использовании следующей команды:

layer1 = K.repeat_elements(layer, A, axis=1)

В результате я получаю следующую ошибку:

AttributeError: у объекта 'Tensor' нет атрибута '_keras_history'

Я считаю, что причина в том, что repeat_elements создает какой-то слой, который не является слоем Keras. Любые альтернативы или идеи о том, как проверить, является ли это слоем Keras или нет?

Answer 1

Вы не можете использовать бэкэнд-функции непосредственно в тензорах, создаваемых слоями Keras, вам нужно обернуть их в слой Lambda с лямбда-функцией или аналогичным образом:

layer1 = Lambda(lambda x: K.repeat_elements(x, A, axis=1))(layer)