def _batch_l2_norm(x: torch.Tensor) -> torch.Tensor:
            """
            Perform a batch L2 norm

            :param x: The inputs to compute the batch L2 norm of
            """
            x_flat = x.reshape(x.size(0), -1)
            return torch.norm(x_flat, dim=1)

            Perform a batch L2 norm

            :param x: The inputs to compute the batch L2 norm of
            
art/estimators/certification/randomized_smoothing/smooth_mix/pytorch.py