I know you asked for the shorter solution, but with the index approach, {⍺[;⍵~⍨⍳⊃⌽⍴⍺]} is 1 character longer but more performant than doing a transpose

Available now: U-Net CNN in APL: Exploring Zero-Framework, Zero-Library Machine Learning by Aaron Hsu (@arcfide) and Rodrigo Girão Serrão (@RGS).