BNReLU2d

class torch.ao.nn.intrinsic.quantized.BNReLU2d(num_features, eps=1e-05, momentum=0.1, device=None, dtype=None) [исходный код]

Модуль BNReLU2d объединяет BatchNorm2d и ReLU.

Мы используем тот же интерфейс, что и torch.ao.nn.quantized.BatchNorm2d.

Переменные:

torch.ao.nn.quantized.BatchNorm2d (то же, что и) –

forward(input) [исходный код]

Применяет объединённые BatchNorm2d и ReLU.

classmethod from_float(mod, use_precomputed_fake_quant=False) [исходный код]

Создаёт квантованный модуль из модуля с плавающей точкой.

classmethod from_reference(bn_relu, output_scale, output_zero_point) [исходный код]

Создаёт квантованный модуль из эталонного модуля.