Why sigma is repeated 10000 times

[EBGU]

Hi,

Your work is really interesting! However, when I run your code, I found a code block that runs very slow:

torsional-diffusion/diffusion/torus.py , line 68

score_norm_ = score(
sample(sigma[None].repeat(10000, 0).flatten()),
sigma[None].repeat(10000, 0).flatten()
).reshape(10000, -1)
score_norm_ = (score_norm_ ** 2).mean(0)

I wonder why sigma is repeated 10000 times? Is there any way to make it faster?
Thanks！