Никто не заинтересован в этом вопросе. После нескольких испытаний я нашел решение. Проблема в том, что, как сообщает Mainak431 в этом репозитории GitHub: ссылка на операции diff и non-diff в тензорном потоке
Существуют дифференцируемые операции TensorFlow и недифференцируемые операции. В записной книжке Colab я использовал, в качестве примера, scatter_nd_update, который нельзя дифференцировать. Поэтому я предлагаю, если вы хотите создать свой собственный пользовательский слой в Keras, чтобы взглянуть на приведенные выше списки, чтобы использовать операции, которые позволяют Keras автоматически дифференцировать вас. В любом случае, я работаю над тем, чтобы как можно больше информировать об этом открытом исследовании topi c. Я помню, что с нейронной сетью «LE GO -ing» граничит, и я точно знаю, что многие из вас заинтересованы в добавлении ваших операций (агрегация или что-то еще) в модель глубокой нейронной сети. Отдельное спасибо Maniak431, я люблю тебя <3 </p>