Revert "Fixed bug w/ batch fft"

This reverts commit 04123fb63022dc6010b1c82d5517897a5c3890e6.
Optimization