Where is torch._C._functions.BatchNorm located?

1

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

1 Like
2

Hi,

This is one of the functions that has been moved to cpp.
It is implemented in this folder.
Especially, the argument parsing is here.
And the definition on how to access elements such as running_mean or running_var is here.

4 Likes