Pytorch使用手册-对您的 PyTorch 模块进行性能分析（专题三十七）

PyTorch 包含一个 Profiler API，它对于识别代码中各种 PyTorch 操作的时间和内存消耗非常有用。Profiler 可以轻松地集成到您的代码中，并且结果可以作为表格打印出来，或者以 JSON 跟踪文件的形式返回。

注意
Profiler 支持多线程模型。Profiler 在与操作相同的线程中运行，但它也会分析可能在另一个线程中运行的子操作。并发运行的 profiler 会被限定在各自的线程中，以防止结果混合。

注意
PyTorch 1.8 引入了新的 API，未来版本中将替代旧的 Profiler API。可以在此页面查看新的 API。

前往这个教程，快速了解 Profiler API的使用方法。

import torch
import numpy as np
from torch import nn