快捷方式

torch.signal.windows.general_cosine

torch.signal.windows.general_cosine(M, *, a, sym=True, dtype=None, layout=torch.strided, device=None, requires_grad=False)[source]

计算一般余弦窗。

一般余弦窗定义如下

wn=i=0M1(1)iaicos(2πinM1)w_n = \sum^{M-1}_{i=0} (-1)^i a_i \cos{ \left( \frac{2 \pi i n}{M - 1}\right)}

该窗口归一化为 1(最大值为 1)。但是,如果 M 为偶数且 symTrue,则 1 不会出现。

参数

M (int) – 窗口的长度。换句话说,返回窗口的点数。

关键字参数
  • a (Iterable) – 与每个余弦函数相关联的系数。

  • sym (bool, 可选) – 如果为 False,则返回适合用于频谱分析的周期性窗口。 如果为 True,则返回适合用于滤波器设计的对称窗口。 默认值:True

  • 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, 可选) – 是否应记录对返回张量的操作。 默认值:False

返回类型

张量

示例

>>> # Generates a symmetric general cosine window with 3 coefficients.
>>> torch.signal.windows.general_cosine(10, a=[0.46, 0.23, 0.31], sym=True)
tensor([0.5400, 0.3376, 0.1288, 0.4200, 0.9136, 0.9136, 0.4200, 0.1288, 0.3376, 0.5400])

>>> # Generates a periodic general cosine window wit 2 coefficients.
>>> torch.signal.windows.general_cosine(10, a=[0.5, 1 - 0.5], sym=False)
tensor([0.0000, 0.0955, 0.3455, 0.6545, 0.9045, 1.0000, 0.9045, 0.6545, 0.3455, 0.0955])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源