Обучаем модель в Керасе с недифференцируемым слоем

Question

Как я могу обучить глубокую нейронную сеть в Кера (Tensorflow) со слоем квантования в середине? т.е. я хочу сделать квантование представления в определенном слое сети (используя векторное квантование) и затем перейти к следующему уровню.

Answer 1

Вы можете использовать лямбда-слой.

def quantize(x):
    # Your vector quantization code

model.add(Lambda(quantize))