如何在YAML文件中计算不同作用域别名的乘积？

问题背景 #

需要让YAML配置中model_parameters.heads_params.in_features等于dataloader_parameters.max_frames（别名&max_frames）与model_parameters.rnn_hidden（别名&rnn_hidden）的乘积，但直接写*max_frames * *rnn_hidden会触发PyYAML的ScannerError，提示Unexpected alias at node end。

核心原因 #

标准YAML语法不支持对别名进行直接算术运算，别名*xxx只能作为独立的节点值存在，无法和运算符组合使用。

可行解决方案 #

1. 自定义PyYAML构造器（推荐） #

通过自定义标签实现算术运算，让PyYAML加载时自动计算乘积：

配置文件示例（config.yaml）

dataloader_parameters:
  max_frames: &max_frames 100
model_parameters:
  rnn_hidden: &rnn_hidden 256
  heads_params:
    in_features: !multiply [*max_frames, *rnn_hidden]

Python加载代码

import yaml

def multiply_handler(loader, node):
    # 解析标签后的序列值，计算乘积
    values = loader.construct_sequence(node)
    result = 1
    for val in values:
        result *= val
    return result

# 注册自定义构造器
yaml.add_constructor('!multiply', multiply_handler)

# 加载配置
with open('config.yaml', 'r') as f:
    config = yaml.safe_load(f)

# 验证结果：输出25600
print(config['model_parameters']['heads_params']['in_features'])

2. 加载后手动计算（最简单） #

如果不需要在YAML中体现计算逻辑，可在Python加载配置后手动赋值：

配置文件示例（config.yaml）

dataloader_parameters:
  max_frames: &max_frames 100
model_parameters:
  rnn_hidden: &rnn_hidden 256
  heads_params:
    in_features: 0  # 占位值

Python处理代码

import yaml

with open('config.yaml', 'r') as f:
    config = yaml.safe_load(f)

# 手动计算并赋值
config['model_parameters']['heads_params']['in_features'] = (
    config['dataloader_parameters']['max_frames'] * config['model_parameters']['rnn_hidden']
)

3. 预处理器替换（不推荐） #

先读取YAML文本，通过字符串替换解析别名数值并计算，再加载。这种方法需要处理复杂的正则匹配，灵活性差，仅适合简单场景。

内容的提问来源于stack exchange，提问作者Eduardo Pacheco