Spec-Zone.ru › PyTorch 1

torch.nn.functional.conv_transpose1d

torch.nn.functional.conv_transpose1d(input, weight, bias=None, stride=1, padding=0, output_padding=0, groups=1, dilation=1) → Tensor

Применяет оператор транспонированной свёртки 1D к входному сигналу, состоящему из нескольких входных плоскостей, иногда называемой «деконволюцией».

Этот оператор поддерживает TensorFloat32.

Подробности и форма вывода см. в ConvTranspose1d.

Примечание

В некоторых случаях, когда используются тензоры на устройстве CUDA и CuDNN, этот оператор может выбрать недетерминированный алгоритм для повышения производительности. Если это нежелательно, вы можете попытаться сделать операцию детерминированной (возможно, с затратами на производительность) установив torch.backends.cudnn.deterministic = True. См. Воспроизводимость для получения дополнительной информации.

Параметры:
  • input – входной тензор формы (минимальная_партія,число_входных_каналов,iW)(\text{minibatch} , \text{in\_channels} , iW)
  • weight – фильтры формы (число_входных_каналов,число_выходных_каналовгруппы,kW)(\text{in\_channels} , \frac{\text{out\_channels}}{\text{groups}} , kW)
  • bias – необязательный bias формы (число_выходных_каналов)(\text{out\_channels}). Значение по умолчанию: None
  • stride – шаг свёртки. Может быть одним числом или кортежем (sW,). Значение по умолчанию: 1
  • padding – dilation * (kernel_size - 1) - padding нулевое заполнение будет добавлено к обеим сторонам каждого измерения на входе. Может быть одним числом или кортежем (padW,). Значение по умолчанию: 0
  • output_padding – дополнительный размер, добавляемый к одной стороне каждого измерения в форме выходных данных. Может быть одним числом или кортежем (out_padW). Значение по умолчанию: 0
  • groups – разделить вход на группы, число_входных_каналов\text{in\_channels} должно быть кратно числу групп. Значение по умолчанию: 1
  • dilation – расстояние между элементами ядра. Может быть одним числом или кортежем (dW,). Значение по умолчанию: 1

Примеры:

>>> inputs = torch.randn(20, 16, 50)
>>> weights = torch.randn(16, 33, 5)
>>> F.conv_transpose1d(inputs, weights)

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

Spec-Zone.ru

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