torch.compiler.assume_constant_result

torch.compiler.assume_constant_result(fn) [source]

Эта функция используется для пометки функции fn как имеющей постоянный результат. Это позволяет компилятору оптимизировать вашу функцию Возвращает ту же функцию fn

Параметры

fn – Функция, которая должна быть помечена как имеющая постоянный результат.

Предупреждение

assume_constant_result может вызвать проблемы безопасности и корректности, torch.compile() не будет пытаться проверить, является ли постоянное предположение истинным или нет