Dimension mismatch in dataloader

New to Julia
1

I have

julia> size(X_train)
(130, 4)

julia> size(y_train)
(130, 3)

but I get a dimension mismatch with

loader = Flux.DataLoader((X_train, y_train), shuffle=true)

((( Oops, I just need to transpose everything. )))

2

yes, in Flux the batch dimension is the last dimension