評價此頁

torch.cuda.nvtx.range_push#

torch.cuda.nvtx.range_push(msg)[原始碼]#

將一個範圍推入巢狀範圍跨度的堆疊。返回開始的範圍的零基深度。

引數

msg (str) – 要與範圍關聯的 ASCII 訊息