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, 可选, 默认值 = -1) –

    要变换到频域的维度的索引。

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

  • bytes_per_sample_hint (int 或 int 列表, 可选, 默认值 = [0]) –

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

    如果指定，则将预先分配位于 GPU 或页锁定主机内存中的操作符输出，以容纳此大小的样本批次。

  • nfft (int, 可选) –

    FFT 的大小。

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

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

    nfft // 2 + 1

    注意

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

  • power (int, 可选, 默认值 = 2) –

    FFT 幅度的指数。

    支持的值为

    • 2 表示功率谱 (real*real + imag*imag)

    • 1 表示复数幅度 (sqrt(real*real + imag*imag))。

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

在本页中