torch.sym_max

torch.sym_max(a, b)

SymInt-aware 的 max 实用工具，避免在 a < b 时分支。与内置的 max()不同，此工具仅适用于 int/float，如果任何参数是 float，则总是提升为 float（与内置的 max 不同，它将忠实地保留输入参数的类型）。