Solved by Eta_C in post #5 
              
              
                What about 
result = torch.cat(
    [x.mean(-1), x.mean(-2),
     x.min(-1)[0], x.min(-2)[0], 
     x.max(-1)[0], x.max(-2)[0]], 
    dim=-1)