stack_compress時のstack次元を可変にする
現在OpenAI実装にあわせて
-1
(= 最後)を stack次元にしているが、指定できるようにしたい。