Hi @ptrblck thanks for the answer. Here is the issue: https://github.com/pytorch/pytorch/issues/40287. Let me know if it needs to be revised, it is my first time creating one.
I also noticed that f.std(true, 0) collapses dimension 1, which is unintuitive, and also f.std(true, 1) collapses dimension 1 still but uses Bessel correction. All in all kinda weird behavior.
I’m actually not blocked on this as it’s a pretty easy to implement function (for my specific case), just pointing it out in case it is interesting to you guys.