torch.compiler.assume_constant_result

torch.compiler.assume_constant_result(fn) [исходный код]

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

Параметры:

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

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

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