PyTorch中Leaky ReLU函数的使用指南

在 PyTorch 中，可以使用 torch.nn.functional 模块中的 leaky_relu 函数来调用 Leaky ReLU。以下是一个示例：

import torch.nn.functional as F

x = torch.randn(10, 5)
y = F.leaky_relu(x, negative_slope=0.01)

其中，x 是输入张量，negative_slope 是一个乘数，用于指定 Leaky ReLU 函数中的负斜率。默认情况下，negative_slope 为 0.01。在上面的示例中，我们将输入张量 x 应用了 Leaky ReLU，并将结果保存在变量 y 中。