Как распечатать значения тензора между слоями в Keras 3? ⇐ Python

[Anonymous]

Я использую Keras 3 с бэкэндом PyTorch.
Я пытаюсь перенести модель, написанную кем-то другим, в другую среду выполнения и хочу сбросить сводную статистику о тензор после каждого слоя, чтобы я мог выяснить, какую операцию я неправильно реализовал в своем порте (вероятно, внимание, лол).
Как вставить операторы печати в модель Keras 3? Все остальные ответы, которые я могу найти, связаны с tf.keras, который, похоже, полностью отличается от того, что я использую. Метода keras.backend.print_tensor() тоже нет.
Я также пытался создать подобную промежуточную модель (для контекста модель, которую я выбираю, — это Moonshine). :
encoder = model.encoder.encoder
encoder_intermediate_model = Model(
inputs=encoder.inputs, outputs=[layer.output for layer in encoder.layers]
)
< /code>
Но попытка запустить эти сбои с расплывчатой ​​ошибкой: < /p>
Traceback (most recent call last):
File "C:\Users\ibiyemi\projects\wellington-ml\moonshine.py", line 764, in
encoder_outputs = encoder_intermediate_model(
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\ibiyemi\projects\wellington-ml\.venv\Lib\site-packages\keras\src\utils\traceback_utils.py", line 122, in error_handler
raise e.with_traceback(filtered_tb) from None
File "C:\Users\ibiyemi\projects\wellington-ml\.venv\Lib\site-packages\torch\nn\modules\module.py", line 1736, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\ibiyemi\projects\wellington-ml\.venv\Lib\site-packages\torch\nn\modules\module.py", line 1747, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
KeyError: "Exception encountered when calling Functional.call().\n\n\x1b[1m2365371176512\x1b[0m\n\nArguments received by Functional.call():\n • inputs=['torch.Tensor(shape=torch.Size([1, 1248, 416]), dtype=float32)', 'torch.Tensor(shape=torch.Size([1]), dtype=int32)']\n • training=None\n • mask=['None', 'None']"


Подробнее здесь: https://stackoverflow.com/questions/793 ... in-keras-3