Source code for horovod.torch.sync_batch_norm ... input.numel() // input.size(1) count = torch.tensor([size]) # calculate mean/invstd for input. mean, ...
確定! 回上一頁