torch.cuda.nvtx.mark

torch.cuda.nvtx.mark(msg)

描述在某個時間點發生的瞬時事件。

參數

msg (str) – 與事件關聯的 ASCII 訊息。