torch.sym_max

torch.sym_max(a, b)

SymInt 感知的 max 工具，它避免了对 a < b 进行分支判断。与内置的 max() 不同，此函数仅适用于 int/float，并且如果任何参数是 float，它总是会提升到 float（与内置的 max() 不同，后者会忠实地保留输入参数的类型）。