[解決済み] Pytorchのテンソルからnumpyの配列への変換

質問

私は pytorch サイズのテンソル torch.Size([4, 3, 966, 1296])

に変換したい。 numpy の配列を以下のコードで作成します。

imgs = imgs.numpy()[:, ::-1, :, :]

どなたか、このコードが何をしているのか説明してください。

解決方法は？

変換したいテンソルの次元は4つである。

[:, ::-1, :, :] 

: は、1次元をそのままコピーして変換すること、3次元、4次元も同様であることを意味します。

::-1 は、2 番目の軸の場合、軸を反転させることを意味します。