ExecutionCUDA

class nvmath.fft.ExecutionCUDA(device_id: int | None = None)

用于为 FFT 对象和包装函数系列 fft(), ifft(), rfft(), 和 irfft() 提供 GPU 执行选项的数据类。

device_id

CUDA 设备序号（如果操作数位于 CPU 上则使用）。设备 0

类型:

int | None

如果未指定， 将 使用 设备 0。

另请参阅

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