I want to create a tensor that looks like this:
[ [0, 1, 2, 3, ...], [0, 1, 2, 3, ...], ... ]
I know I can do this with a loop and
tc.stack([tc.arange(0, L) for i in range(B)])
However, this method can be inefficient because of big python loops. I’m wondering is there a way to achieve this without loops?