Проблема с пользовательской функцией активации — TypeError: аргумент типа «Bind» не повторяется. ⇐ Python

[Anonymous]

Я создаю автокодировщик и хочу закодировать свои значения в логическую матрицу. Я использую свою собственную функцию активации шага на последнем уровне, как показано ниже:

Код: Выделить всё

@tf.custom_gradient
def binary_activation(x):

ones = tf.ones(tf.shape(x), dtype=x.dtype.base_dtype)
zeros = tf.zeros(tf.shape(x), dtype=x.dtype.base_dtype)

def grad(dy):
return ...  # TODO define gradient
return keras.backend.switch(x > 0.5, ones, zeros), grad

Однако после запуска нескольких эпизодов появится ошибка:
[img]https://i.stack.imgur. com/rA3hV.png[/img]
Может ли кто-нибудь помочь мне с этой проблемой?
Спасибо,

Подробнее здесь: https://stackoverflow.com/questions/783 ... ype-bind-i