torch 画图 一张图画多个子图

可以使用 `matplotlib` 库来实现在一张图中画多个子图的功能。下面是一个示例代码：

import matplotlib.pyplot as plt
import torch

# 生成数据
x = torch.linspace(-10, 10, 100)
y1 = torch.sin(x)
y2 = torch.cos(x)
y3 = torch.tan(x)

# 创建画布和子图
fig, (ax1, ax2, ax3) = plt.subplots(3, 1, figsize=(8, 10))

# 绘制子图1
ax1.plot(x, y1, 'r-', linewidth=2)
ax1.set_title('y=sin(x)')

# 绘制子图2
ax2.plot(x, y2, 'b-', linewidth=2)
ax2.set_title('y=cos(x)')

# 绘制子图3
ax3.plot(x, y3, 'g-', linewidth=2)
ax3.set_title('y=tan(x)')

# 调整子图之间的距离和位置
plt.subplots_adjust(hspace=0.5, top=0.95, bottom=0.05)

# 显示图像
plt.show()

在这个示例中，我们使用 `subplots` 函数创建了一个包含三个子图的画布，每个子图显示一条不同的曲线。`set_title` 函数用于设置子图的标题，`subplots_adjust` 函数用于调整子图之间的距离和位置。最后，调用 `show` 函数显示图像。