LazyConv2d

class torch.nn.modules.conv.LazyConv2d(out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros', device=None, dtype=None) [source]

Модуль torch.nn.Conv2d с отложенной инициализацией аргумента in_channels.

Аргумент in_channels объекта Conv2d, который определяется по значению input.size(1). Атрибуты, инициализируемые отложенно: weight и bias.

Дополнительную документацию об отложенно инициализируемых модулях и их ограничениях см. в torch.nn.modules.lazy.LazyModuleMixin.

Параметры:

• out_channels (int) – Количество каналов, создаваемых свёрткой
• kernel_size (int or tuple) – Размер ядра свёртки
• stride (int or tuple, необязательный) – Шаг свёртки. По умолчанию: 1
• padding (int or tuple, необязательный) – Нулевое дополнение, добавляемое с обеих сторон входных данных. По умолчанию: 0
• dilation (int or tuple, необязательный) – Расстояние между элементами ядра. По умолчанию: 1
• groups (int, необязательный) – Количество блокируемых соединений между входными и выходными каналами. По умолчанию: 1
• bias (bool, необязательный) – Если True, к выходным данным добавляется обучаемое смещение. По умолчанию: True
• padding_mode (str, необязательный) – 'zeros', 'reflect', 'replicate' или 'circular'. По умолчанию: 'zeros'

См. также

torch.nn.Conv2d и torch.nn.modules.lazy.LazyModuleMixin

cls_to_become [source]

псевдоним Conv2d