Пользовательский уровень пула для CNN ⇐ Python

[Anonymous]

Я хочу создать собственный уровень пула 2x2 для моей архитектуры CNN с помощью этой функции с помощью Keras

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

def min_max_cantor_pool2d(x):
def cantor_pairing(a, b):
return (a + b) * (a + b + 1) // 2 + b

def minmax_norm(x, min_value, max_value):
return (x - min_value) / (max_value - min_value)

cantor_x = cantor_pairing(x[:, :, :, 0], x[:, :, :, 1])
norm_x = minmax_norm(cantor_x, 0, 510*511*0.5+255)

return K.expand_dims(norm_x, axis=-1)

я хочу, чтобы функция получала этот ввод:

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

[a, b

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

c, d]

и возврат

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

cantor_pairing(cantor_pairing(a, b), cantor_pairing(c, d))

Но почему-то всегда выдает ошибку. Может ли кто-нибудь помочь мне с этой проблемой?

Подробнее здесь: https://stackoverflow.com/questions/790 ... er-for-cnn