torch.sym_max

torch.sym_max(a, b)

SymInt 感知的 max 实用程序，避免对 a < b 进行分支。与 builtins.max() 不同，此函数仅适用于整数/浮点数，并且如果任何参数为浮点数，则始终提升为浮点数（与 builtins.max 不同，后者将忠实地保留输入参数的类型）。