def _normalize(z: torch.Tensor) -> torch.Tensor:
    """Helper function to create batches of mean 0 and std 1."""
    return F.batch_norm(
        z,
        running_mean=None,
        running_var=None,
        weight=None,
        bias=None,
        training=True,
    )
Helper function to create batches of mean 0 and std 1.
lightly/loss/barlow_twins_loss.py