torch.cuda.nvtx.range

torch.cuda.nvtx.range(msg, *args, **kwargs)

Context manager / decorator，在範圍的開頭推送一個 NVTX 範圍，並在結尾彈出它。 如果給出了額外的參數，它們會作為參數傳遞給 msg.format()。

參數

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