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！