torch.nn.functional.relu6

torch.nn.functional.relu6(input, inplace=False) → Tensor

Applies the element-wise function $\text{ReLU6}(x) = \min(\max(0,x), 6)$.

See ReLU6 for more details.

Return type

Tensor