如何修改LoggingTensorHook格式化器？自定义其打印内容方法

嘿，我来帮你解决这两个关于TensorFlow LoggingTensorHook的问题：

LoggingTensorHook默认会用简单格式打印张量的全部值，但你可以通过它的formatter参数传入自定义函数，完全控制输出的格式。

步骤很简单：

1. 定义一个格式化函数，它接收一个字典作为输入（键是张量名称，值是对应张量的numpy数组）
2. 在函数里按照你想要的方式处理这些值，返回最终要打印的字符串
3. 创建LoggingTensorHook时把这个函数传给formatter参数

举个实用的例子，比如我们只想打印每个张量的均值、形状和标量值：

import tensorflow as tf

def custom_log_formatter(tensor_values):
    log_output = []
    for tensor_name, tensor_val in tensor_values.items():
        # 针对标量张量直接打印值，针对多维张量打印均值和形状
        if tensor_val.ndim == 0:
            log_output.append(f"{tensor_name}: {tensor_val:.4f}")
        else:
            log_output.append(f"{tensor_name}: mean={tensor_val.mean():.4f}, shape={tensor_val.shape}")
    return " | ".join(log_output)

# 创建Hook时传入自定义格式化器
logging_hook = tf.train.LoggingTensorHook(
    tensors=["train_loss", "train_acc", "conv1_weights"],
    every_n_iter=100,
    formatter=custom_log_formatter
)

这样每次触发Hook时，就会按照你定义的格式输出，而不是默认的全量张量值。

针对你提到的CNN标准实现，默认的Hook可能会打印指定张量的全部值，要修改它只输出你需要的内容，有两种思路：

思路1：只追踪你需要的张量 #

如果原来的Hook追踪了太多张量，第一步可以直接在tensors参数里只保留你关心的（比如损失、准确率），这样Hook就只会打印这些张量，不会输出无关内容：

# 假设教程里的损失和准确率张量分别是loss和accuracy
logging_hook = tf.train.LoggingTensorHook(
    tensors={"训练损失": loss, "训练准确率": accuracy},
    every_n_iter=50
)

思路2：自定义格式化器控制单张量的输出内容 #

如果已经指定了要追踪的张量，但不想打印全量值（比如多维权重张量，你只想看它的统计量），就用问题1里的自定义格式化器方法。比如针对CNN教程，我们可以专门写一个格式化函数：

def cnn_training_formatter(tensor_vals):
    log_str_parts = []
    for name, val in tensor_vals.items():
        if "损失" in name:
            # 损失是标量，保留4位小数
            log_str_parts.append(f"{name}: {val[0]:.4f}")
        elif "准确率" in name:
            # 准确率转成百分比格式
            log_str_parts.append(f"{name}: {val[0]*100:.2f}%")
        elif "权重" in name:
            # 权重张量只打印均值和标准差
            log_str_parts.append(f"{name}: mean={val.mean():.4f}, std={val.std():.4f}")
    return " | ".join(log_str_parts)

# 替换教程里的Hook定义
logging_hook = tf.train.LoggingTensorHook(
    tensors={"训练损失": loss, "训练准确率": accuracy, "卷积层1权重": conv1_weights},
    every_n_iter=50,
    formatter=cnn_training_formatter
)

这样修改后，Hook就只会输出你指定的内容，不会打印庞大的全量张量值了。

内容的提问来源于stack exchange，提问作者Dario Z.