def _tf_fftn(self, x, fft_length=None, axes=None, norm=None, feed_dict=None):
    with self.cached_session() as sess:
      return sess.run(
          fft_ops.rfftnd(x, fft_length=fft_length, axes=axes, norm=norm),