rnd = rnd * (1 - wall.clamp(max=1))
rnd = torch.rand(nb, height, width)
rnd = rnd * (1 - wall.clamp(max=1))
rnd = torch.rand(nb, height, width)
coins = torch.zeros(nb, T, height, width, dtype=torch.int64)
rnd = rnd * (1 - wall.clamp(max=1))
for k in range(nb_coins):
coins = torch.zeros(nb, T, height, width, dtype=torch.int64)
rnd = rnd * (1 - wall.clamp(max=1))
for k in range(nb_coins):