torch.sym_max

torch.sym_max(a, b)

SymInt 感知的 max 工具，它避免了對 a < b 進行分支判斷。與內建的 max() 不同，此函式僅適用於 int/float，並且如果任何引數是 float，它總是會提升到 float（與內建的 max() 不同，後者會忠實地保留輸入引數的型別）。