parameters: (self, x)
return_statement: return _FFT_OP[self.block_depth](x_complex)
docstring: FFT along the last self.block_depth dimensions of x.
  
      Args:
        x: `Tensor` with floating or complex `dtype`.
          Should be in the form returned by self._vectorize_then_blockify.
  
      Returns:
        `Tensor` with `dtype` `complex64`.
docstring_summary: FFT along the last self.block_depth dimensions of x.
docstring_tokens: FFT along the last self . block_depth dimensions of x .
function: def _fft(self, x):
      """FFT along the last self.block_depth dimensions of x.
  
      Args:
        x: `Tensor` with floating or complex `dtype`.
          Should be in the form returned by self._vectorize_then_blockify.
  
      Returns:
        `Tensor` with `dtype` `complex64`.
      """
      x_complex = _to_complex(x)
      return _FFT_OP[self.block_depth](x_complex)
function_tokens: def _fft ( self , x ) : x_complex = _to_complex ( x ) return _FFT_OP [ self . block_depth ] ( x_complex )