torch.nn.functional.leaky_relu

torch.nn.functional.leaky_relu(input, negative_slope=0.01, inplace=False) → Tensor [исходный код]

Применяет поэлементно функцию $\text{LeakyReLU}(x) = \max(0, x) + \text{negative\_slope} * \min(0, x)$

Подробнее см. LeakyReLU.

Тип возвращаемого значения:

Tensor