torch.compiler.assume_constant_result

torch.compiler.assume_constant_result(fn)

此函式用於標記函式 fn 的結果是常量。這允許編譯器最佳化掉您的函式。返回 相同的函式 fn

引數

fn – 要標記為具有常量結果的函式。

警告

assume_constant_result 如果不正確，可能導致安全和健全性問題，torch.compile() 將不會嘗試驗證常量假設是否屬實。