torch.sym_max

torch.sym_max(a: IntLikeType, b: IntLikeType) → IntLikeType [source]

torch.sym_max(a:IntLikeType|FloatLikeType, b:IntLikeType|FloatLikeType) → FloatLikeType

Утилита для вычисления максимума с поддержкой SymInt, которая позволяет избежать ветвления по условию a < b. В отличие от встроенной функции max(), она работает только с int/float и всегда преобразует результат к float, если хотя бы один аргумент имеет тип float (в отличие от встроенной функции max, которая сохраняет тип входного аргумента).