Solved by googlebot in post #2 
              
              
                What output shape do you want? Generally, non-rectangular slicing requires gather(), i.e. specifying indexes explicitly. Or a loop.