快捷方式

torch.signal.windows.cosine

torch.signal.windows.cosine(M, *, sym=True, dtype=None, layout=torch.strided, device=None, requires_grad=False)[源代码]

计算一个具有简单余弦波形的窗口,遵循与 SciPy 相同的实现。此窗口也称为正弦窗口。

余弦窗口定义如下

wn=sin(π(n+0.5)M)w_n = \sin\left(\frac{\pi (n + 0.5)}{M}\right)

此公式与典型的余弦窗口公式的不同之处在于分子中包含 0.5 项,该项会移动样本位置。此调整导致窗口以非零值开始和结束。

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

参数

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

关键字参数
  • 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 (布尔值, 可选) – 是否应记录返回张量的操作。默认值:False

返回类型

张量

示例

>>> # Generates a symmetric cosine window.
>>> torch.signal.windows.cosine(10)
tensor([0.1564, 0.4540, 0.7071, 0.8910, 0.9877, 0.9877, 0.8910, 0.7071, 0.4540, 0.1564])

>>> # Generates a periodic cosine window.
>>> torch.signal.windows.cosine(10, sym=False)
tensor([0.1423, 0.4154, 0.6549, 0.8413, 0.9595, 1.0000, 0.9595, 0.8413, 0.6549, 0.4154])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源