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

Возвращает:

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

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

Тензор