torch.compiler.assume_constant_result

torch.compiler.assume_constant_result(fn)

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

参数

fn – 要标记为具有常量结果的函数。

警告

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