torch.cuda.nvtx.mark

torch.cuda.nvtx.mark(msg)[源代码]

描述一个在某个时刻发生的瞬间事件。

参数

msg (str) – 事件相关的 ASCII 消息。