Как записать значения тензора во время тренировки в Керасе?

Question

У меня есть настроенный слой в Keras, и я хочу отслеживать определенный тензор во время обучения (т. Е. Оценивать этот тензор за эпоху и записывать значение в список).Что я должен сделать, чтобы добиться этого?

Кусок кода в качестве примера:

class customizedLayer(Layer):
    def __init__():
        ...
    def build():
        ...
    def call():
        tensor_1 = ...
        tensor_2 = ...
        return ...

output = customizedLayer(input)
model = Model(input=X, output=Y)
model.compile(...)
model.fit(...)

что я должен сделать, чтобы получить значения tenor_1 или тензор_2 показанный выше?

Большое спасибо.

Answer 1

Вы можете реализовать объект обратного вызова для сохранения этих значений в файл каждую эпоху, я думаю. Если у меня есть что-то более конкретное, я говорю вам.