torch.compiler.assume_constant_result

torch.compiler.assume_constant_result(fn)

此函数用于标记函数 fn 具有恒定结果，这允许编译器优化掉您的函数。返回相同的函数 fn

参数:

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

警告

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