I’m going to use NVIDIA’s VPI(Vision Programming Interface) for the acceleration of FFT&IFFT in the Jetson Xavier NZ module.
But it seems that FFT in VPI module only supports ‘VPI.Image’ format input.
Is it possible to do FFT operation of VPI library with a pytorch embedding tensor (which has a larger dimension than 3), not an image?
(e.g. pytorch tensor of shape (30, 30, 256), which is H, W, C)
If not, is there a way to do FFT faster even if it’s not a VPI library in Jetson Nano?