I see, yes that is correct, thanks for the clarification (I think you meant m(embedding2).mean(-1), m(embedding2).std(-1, unbiased=False)  though).
so do you know when we track_running_stats=False what is computed in the m(embedding1)?