torch.sym_max

torch.sym_max(a, b)

SymInt 感知的最大值工具，避免對 a < b 進行分支。 與 builtins.max() 不同，這僅適用於 int/float，並且如果任何參數為 float，它總是會提升為 float（與 builtins.max 不同，它會忠實地保留輸入參數的類型）。