LazyConv2d

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

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

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

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

Параметры:

• out_channels (int) – Количество каналов, создаваемых свёрткой
• kernel_size (int или tuple) – Размер ядра свёртки
• stride (int или tuple, необязательно) – Шаг свёртки. По умолчанию: 1
• padding (int или tuple, необязательно) – Дополнение нулями с обеих сторон входных данных. По умолчанию: 0
• dilation (int или 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 [исходный код]

псевдоним Conv2d