Conv3d

class torch.ao.nn.qat.Conv3d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros', qconfig=None, device=None, dtype=None) [исходный код]

Модуль Conv3d с подключёнными модулями FakeQuantize для весов, используемый для обучения с учётом квантования.

Мы используем тот же интерфейс, что и torch.nn.Conv3d; документацию см. на странице https://pytorch.org/docs/stable/nn.html?highlight=conv3d#torch.nn.Conv3d.

Подобно torch.nn.Conv3d, модули FakeQuantize инициализируются значениями по умолчанию.

Переменные:

weight_fake_quant – модуль фиктивного квантования для весов