如何在tensorflow中运行model.fit（）时输出一些数据？

我想在model.fit运行期间而不是之前打印张量的值和/或形状。 在PyTorch中，我可以将printinput.shape语句放入model.forward函数中。tensorflow中有类似的东西吗

谢谢

您可以将回调对象传递给model.fit方法，然后在拟合过程中的不同阶段执行操作

类MyCustomCallbacktf.keras.callbacks.Callback： 列车上的def批次自身、批次、日志=无： 打印“培训：批处理{}从{}开始”。formatbatch，datetime.datetime.now.time 列车上的def\u批次\u结束自身、批次、日志=无： 打印“培训：批处理{}在{}结束”。formatbatch，datetime.datetime.now.time def on_test_batch_beginself，batch，logs=无： 打印“评估：批处理{}从{}开始”。formatbatch，datetime.datetime.now.time def on_test_batch_endself，batch，logs=无： 打印“评估：批处理{}在{}结束”。formatbatch，datetime.datetime.now.time 模型=获取模型 model.fitx\u train，y\u train，回调=[MyCustomCallback]