cupy.cuda.cufft.Plan1d.fftでfftを行ったときのプロットを行いたい

Question

現在、Pythonのライブラリであるcupyを用いて以下のページを参考にfftを行っています。
https://qiita.com/not522/items/003b06d7081b05a52f89

単純にnumpyで行うのと同じ要領で以下のようにcupyを用いてfftを行うとoutの部分にx_gpu値がfftされた配列が戻ってきます。
out=cupy.fft.fft(x_gpu)

しかしさらなる高速化のために参考ページのcupy.cuda.cufft.Plan1dの項目にあるplanを先に設定する手順の場合に、以下のようにfftを行った場合はoutがNoneになります。
out=plan.fft(x_gpu, y_gpu, cufft.CUFFT_FORWARD)

以上のようにcupy.cuda.cufft.Plan1d.fftを行ったときに引数に対してfftを行った戻り値を受け取るにはどうしたらいいでしょうか。

Answer 1

y_gpu = plan.get_output_array(x_gpu) は、結果が入る領域を確保します。
plan.fft(x_gpu, y_gpu, cufft.CUFFT_FORWARD) により
fft の結果は y_gpu に入ります。print(y_gpu) で結果を出力できます。
plan.fft には戻り値がないので out = plan.fft(..) としても out は None です。