nvidia.dali.fn.power_spectrum#

nvidia.dali.fn.power_spectrum(__input, /, *, axis=-1, bytes_per_sample_hint=[0], nfft=None, power=2, preserve=False, device=None, name=None)#

计算信号的功率谱。

支持的后端

  • ‘cpu’

参数:

__input (TensorList) – 操作符的输入。

关键字参数:

  • axis (int, optional, default = -1) –

    要转换为频域的维度的索引。

    默认情况下，选择最后一个维度。

  • bytes_per_sample_hint (int or list of int, optional, default = [0]) –

    输出大小提示，以字节/样本为单位。

    如果指定，则驻留在 GPU 或页锁定主机内存中的操作符输出将被预先分配以适应此大小的样本批次。

  • nfft (int, optional) –

    FFT 的大小。

    默认情况下，选择 nfft 以匹配变换轴中数据的长度。

    输出中创建的 bin 的数量使用以下公式计算

    nfft // 2 + 1

    注意

    输出仅表示频谱的正半部分。

  • power (int, optional, default = 2) –

    FFT 幅度的指数。

    支持的值为

    • 2 用于功率谱 (real*real + imag*imag)

    • 1 用于复数幅度 (sqrt(real*real + imag*imag))。

  • preserve (bool, optional, default = False) – 阻止操作符从图中移除，即使其输出未使用。

在本页