ここに飛ばされたんですが、いろいろ試してみましたが、
<class 'tuple'>: (1, 1, 320, 320) この形の配列を
<class 'tuple'>: (320, 320, 1) にするやり方が分かりません。
forループで要素を入れ替えると、約1秒もかかります。