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

[Anonymous]

Я использую Keras 3 с бэкэнд Pytorch (я не могу заставить TF Backend по какой -то причине, и я никогда не использовал JAX). < /p>
i ' M Попытка перенести модель, написанную кем -то другим в другое время выполнения, и я хочу сбросить сводную статистику о тензоре после каждого уровня, чтобы я мог выяснить, какую операцию я не неправильно реализовал в своем порту (вероятно, внимание, лол).
Как вставить операторы печати в модель Keras 3 ?? Все остальные ответы, которые я могу найти, связаны с tf.keras , что, кажется, полностью отличается от того, что я использую. Не существует метода keras.backend.print_tensor () . : < /p>
encoder = model.encoder.encoder
encoder_intermediate_model = Model(
inputs=encoder.inputs, outputs=[layer.output for layer in encoder.layers]
)

Но при попытке запустить это происходит сбой с непонятной ошибкой:
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