diffusion_models.models.nn.mean_flat¶ diffusion_models.models.nn.mean_flat(tensor)[source]¶ Take the mean over all non-batch dimensions.