LazyConv3d

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

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

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

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

Параметры:

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

См. также

torch.nn.Conv3d и torch.nn.modules.lazy.LazyModuleMixin

cls_to_become [исходный код]

псевдоним Conv3d