ReLU6

class torch.ao.nn.quantized.ReLU6(inplace=False) [исходный код]

Применяет поэлементную функцию:

$\text{ReLU6}(x) = \min(\max(x_0, x), q(6))$, где $x_0$ — zero_point, а $q(6)$ — квантованное представление числа 6.

Параметры:

inplace (bool) – при необходимости операция может выполняться на месте. По умолчанию: False

Форма:

• Входные данные: $(N, *)$, где * означает любое количество дополнительных измерений
• Выходные данные: $(N, *)$, форма совпадает с формой входных данных

Примеры:

>>> m = nn.quantized.ReLU6()
>>> input = torch.randn(2)
>>> input = torch.quantize_per_tensor(input, 1.0, 0, dtype=torch.qint32)
>>> output = m(input)