This loop is always printing the distance values, and then end, when finished iterating all data.
      ne=ne.detach().cpu().numpy()           
      naa=naa.detach().cpu().numpy()           
      de=0
      for ne, naa in zip(ne, naa):          
        de+=distance.euclidean(naa,ne)
      print(de/size_batch)

I want to calculate the mean of all printed values, it is clear now what i want to do?
The print, prints values like: 3.21412
3.124124
3.112412
3.42121
And i want to calculate the mean of all this printed values.