torch.compiler.assume_constant_result

torch.compiler.assume_constant_result(fn)

此函式用於將函式 fn 標記為具有常數結果。這允許編譯器優化掉您的函式。 傳回：與函式 fn 相同的函式。

參數

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

警告

如果 assume_constant_result 無效，可能會導致安全性和健全性問題，torch.compile() 不會嘗試驗證常數假設是否成立