目錄

快捷方式

torch.cuda.nvtx.range¶

torch.cuda.nvtx.range(msg, *args, **kwargs)[source][source]¶

一個上下文管理器/裝飾器，在其作用域開始時壓入一個 NVTX 範圍，並在結束時彈出。如果提供了額外引數，它們將作為引數傳遞給 msg.format()。

引數: msg (str) – 與範圍關聯的訊息

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源