def _fft_out_chunks(a, n, axis):
 if (n is None):
  return a.chunks
 chunks = list(a.chunks)
 chunks[axis] = (n,)
 return chunks

def _fft_out_chunks ( a , n , axis ) : if ( n is None ) : return a . chunks chunks = list ( a . chunks ) chunks [ axis ] = ( n , ) return chunks
for computing the output chunks of fft and ifft .