torch.sym_max

torch.sym_max(a, b)[源代码]

SymInt感知的max工具，在a < b时不进行分支操作。与builtins.max()不同，此函数仅适用于int和float类型，并且如果任何参数是浮点数，则始终转换为浮点数（而builtins.max会保留输入参数的原始类型）。