Spec-Zone.ru › PyTorch 2

LazyConv1d

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

A модуль с torch.nn.Conv1d с ленивой инициализацией аргумента in_channels модуля Conv1d, который определяется из 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.Conv1d и torch.nn.modules.lazy.LazyModuleMixin

cls_to_become

псевдоним Conv1d

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

Spec-Zone.ru

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