numpyで行列計算の短縮化の方法を教えてください

numpyで行列の計算をしているのですが、下記の方法を短縮できるコードはどのようなものが考えられますか？

行列A[2,2,2]の要素を持つときにこの行列をB[2,4]の行列に置き換えたいです。
行列Bは、行列Aの後半部分の2×2をまとめてあります。

つまり、

B[0,0]=A[0,0,0]
B[0,1]=A[0,0,1]
B[0,2]=A[0,1,0]
B[0,3]=A[0,1,1]

B[1,0]=A[1,0,0]
B[1,1]=A[1,0,1]
B[1,2]=A[1,1,0]
B[1,3]=A[1,1,1]

としたBを作りたいです。
要素数が多くなると、このような計算をするのが困難で、簡単にできる方法があれば教えてください。
よろしくお願いします。