torch.compiler.assume_constant_result

torch.compiler.assume_constant_result(fn)

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

引數

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

警告

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