torch.bartlett_window

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

Функция окна Бартлетта.

$$w[n] = 1 - \left| \frac{2n}{N-1} - 1 \right| = \begin{cases} \frac{2n}{N - 1} & \text{if } 0 \leq n \leq \frac{N - 1}{2} \\ 2 - \frac{2n}{N - 1} & \text{if } \frac{N - 1}{2} < n < N \\ \end{cases},$$

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

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

Примечание

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

Параметры

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

Ключевые аргументы

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

Возвращает

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

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

Тензор