ExecutionCPU

class nvmath.fft.ExecutionCPU(num_threads: int | None = None)

一个数据类，用于为 FFT 对象和包装函数族 fft(), ifft(), rfft(), 和 irfft() 提供 CPU 执行选项。

num_threads

用于执行计划的 CPU 线程数。 如果未指定，则默认为进程可用的 CPU 核心数。

类型:

int | None

另请参阅

ExecutionCUDA, FFT, fft(), ifft(), rfft(), 和 irfft()。