Spec-Zone.ru › PyTorch 2

LazyConvTranspose2d

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

A torch.nn.ConvTranspose2d module with lazy initialization of the in_channels argument of the ConvTranspose2d that is inferred from the input.size(1). The attributes that will be lazily initialized are weight and bias.

Check the torch.nn.modules.lazy.LazyModuleMixin for further documentation on lazy modules and their limitations.

Parameters
  • out_channels (int) – Количество каналов, производимых свёрткой
  • kernel_size (int or tuple) – Размер ядра свёртки
  • stride (int or tuple, optional) – Шаг свёртки. По умолчанию: 1
  • padding (int or tuple, optional) – dilation * (kernel_size - 1) - padding нулевое заполнение будет добавлено к обеим сторонам каждой размерности ввода. По умолчанию: 0
  • output_padding (int or tuple, optional) – Дополнительный размер, добавляемый к одной стороне каждой размерности в форме вывода. По умолчанию: 0
  • groups (int, optional) – Количество заблокированных соединений от входных каналов к выходным каналам. По умолчанию: 1
  • bias (bool, optional) – Если True, добавляет обучаемый сдвиг к выводу. По умолчанию: True
  • dilation (int or tuple, optional) – Расстояние между элементами ядра. По умолчанию: 1

См. также

torch.nn.ConvTranspose2d и torch.nn.modules.lazy.LazyModuleMixin

cls_to_become

alias of ConvTranspose2d

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

Spec-Zone.ru

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