torch.compiler.assume_constant_result

torch.compiler.assume_constant_result(fn)[源代码]

此函数用于标记函数 fn 的结果为常量，从而让编译器对其进行优化。

参数

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

警告

assume_constant_result 如果无效可能会引发安全性和正确性问题，而torch.compile() 不会试图验证该常量假设是否成立。