Yes, it should dispatch into mean.
If I remember it correctly, this call was missing for memory_format=torch.channels_last workloads and was added before 1.7, so I would recommend to use the latest PyTorch version (as always  ).