Hi,
In nn/functional.py,
f = torch._C._functions.BatchNorm(running_mean, running_var, training, momentum, eps, torch.backends.cudnn.enabled)
Can I get the current value of running_mean and running_var from this call? If yes, can I also get the current mini-batch mean and var.
I am looking where this function comes from so that I can figure it out. Any pointers would be most helpful.
Regards
Nabarun