torch.sym_max

torch.sym_max(a, b)

SymInt 感知的 max 实用程序，避免了在 a < b 上进行分支。与 builtins.max() 不同，这仅适用于 int/float，并且如果任何参数是 float，它总是提升为 float（与 builtins.max 不同，后者会忠实地保留输入参数的类型）。