pytorch画relu

PyTorch中的ReLU函数可以使用以下代码进行绘制：

import torch
import matplotlib.pyplot as plt

x = torch.arange(-5., 5., 0.1)
y = torch.relu(x)

plt.plot(x.numpy(), y.numpy())
plt.show()

这段代码会生成一个以x为横轴，y为纵轴的ReLU函数图像。其中，torch.relu()函数是PyTorch中的ReLU函数，可以对输入进行ReLU操作。plt.plot()函数用于绘制图像，plt.show()函数用于显示图像。