I see you're just moving this line, but while I remember we should change this to `.transpose(2, 0, 1)`. No need (and a bit confusing) to factor our permutations. And if you like, it's more succinct to switch to BGR after the permutation.