Как распечатать выходной вектор из капсульного слоя?

Question

Я работаю с капсулярными сетями.Мой справочный документ здесь .В статье говорится, что первичный слой капсулы генерирует 8D вектор параметра позы.Мой вопрос: можем ли мы получить вектор из вывода слоя и напечатать этот 8D вектор?

Answer 1

Поскольку вопрос помечен с помощью keras, я предполагаю, что модель построена с использованием библиотеки keras.

Можем ли мы получить вектор из вывода слоя и напечатать этот 8D вектор?

Да, вы можете.Сначала вы должны получить доступ к входам модели и выходу слоя капсулы.

inputs = model.input
outputs = model.layers[index_of_the_caps_layer].output

Keras предоставляет функцию keras.backend.function для запуска графика вычислений.Вам нужно передать массив numpy для ввода заполнителя модели

import keras.backend as K 
input_values # Input values to the model 
function = K.function([inputs, K.learning_phase()], outputs )
caps_output = function([input_values, 1.0])
print(caps_output)