torch.cuda.nvtx.range

torch.cuda.nvtx.range(msg, *args, **kwargs) [исходный код]

Менеджер контекста / декоратор, который открывает диапазон NVTX в начале области действия и закрывает его в конце. Если переданы дополнительные аргументы, они используются в качестве аргументов для msg.format().

Параметры:

msg (str) – сообщение, связанное с диапазоном