Random Grid Shuffle の実装

ある画像をいくつかのブロックに分割し、ランダムに並び替えたい時があったのでメモ。 albumentationsにもRandomGridShuffleクラスがあり、似たようなことはできるのですがこちらは各ブロックのフリップが出来なそうだったので自分で実装しました。もっと効率的に書けそうな気はするので改善の余地あり。