Proposed new feature or change:

x.repeat(B, 0).repeat(H * W, 3)

In this case, instead of calling repeat twice, I'd love to call it a single time with something along the lines of

x.repeat(
    repeats=(B, H * W),
    axis=(0, 3)
)

or by combining the number of repetitions with the axis like so:

x.repeat((B, 0), (H * W, 3))