torch.sym_max

torch.sym_max(a, b)

用于计算最大值的 SymInt 感知工具，它避免了对 a < b 进行分支判断。与内置的 builtins.max() 不同，它仅适用于 int/float 类型，并且如果任一参数是 float，则总是提升为 float 类型（与 builtins.max 不同，builtins.max 会忠实地保留输入参数的类型）。