torch.hann_window

torch.hann_window(window_length, periodic=True, *, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

Функция окна Ханна.

$$w[n] = \frac{1}{2}\ \left[1 - \cos \left( \frac{2 \pi n}{N - 1} \right)\right] = \sin^2 \left( \frac{\pi n}{N - 1} \right),$$

где $N$ — полный размер окна.

Входной параметр window_length — положительное целое число, задающее размер возвращаемого окна. Флаг periodic определяет, отсекает ли возвращаемое окно последнее повторяющееся значение симметричного окна, чтобы его можно было использовать как периодическое окно с такими функциями, как torch.stft(). Поэтому, если periodic имеет значение true, $N$ в приведённой выше формуле фактически равно $\text{window\_length} + 1$. Кроме того, всегда выполняется равенство torch.hann_window(L, periodic=True) и torch.hann_window(L + 1, periodic=False)[:-1]).

Примечание

Если window_length $=1$, возвращаемое окно содержит единственное значение 1.

Параметры:

• window_length (int) – размер возвращаемого окна
• periodic (bool, необязательный) – если True, возвращает окно для использования в качестве периодической функции. Если False, возвращает симметричное окно.

Именованные аргументы:

• dtype (torch.dtype, необязательно) – требуемый тип данных возвращаемого тензора. По умолчанию: если None, используется глобальное значение по умолчанию (см. torch.set_default_dtype()). Поддерживаются только типы с плавающей точкой.
• layout (torch.layout, необязательно) – требуемый формат возвращаемого тензора окна. Поддерживается только torch.strided (плотный формат).
• device (torch.device, необязательно) – требуемое устройство для возвращаемого тензора. По умолчанию: если None, используется текущее устройство для типа тензора по умолчанию (см. torch.set_default_device()). device будет соответствовать CPU для типов тензоров CPU и текущему устройству CUDA для типов тензоров CUDA.
• requires_grad (bool, необязательный) – следует ли autograd записывать операции над возвращаемым тензором. По умолчанию: False.

Возвращает:

Одномерный тензор размера $(\text{window\_length},)$, содержащий окно

Тип возвращаемого значения:

Tensor