Spec-Zone.ru › PyTorch 2

LazyConv3d

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

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

См. torch.nn.modules.lazy.LazyModuleMixin для дополнительной документации по отложенным модулям и их ограничениям.

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

См. также

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

cls_to_become

Псевдоним для Conv3d

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.nn.LazyConv3d.html

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API