I was wondering how to repeat a 2D tensor
let’s say
a = [
 [x1, x2, x3],
 [y1, y2, y3]
]

to this tensor
[
  [x1,x1,x1,x2,x2,x2,x3,x3,x3],
  [y1,y1,y1,y2,y2,y2,y3,y3,y3]
]


I can’t implement this via torch.cat(),for now, I can only use torch.gather to achieve this, but it’s kind of complicated, is there any way easier?