torch.compiler.assume_constant_result

torch.compiler.assume_constant_result(fn)

此函数用于将一个函数 fn 标记为具有常量结果。 这允许编译器优化掉您的函数 返回相同的函数 fn

Parameters

fn – 要标记为具有恒定结果的函数。

警告

assume_constant_result 如果无效可能会导致安全性和稳定性问题，torch.compile() 将不会尝试验证常量假设是否为真