code: def check_so2_z_shape(z: Tensor) -> None:
      z_shape = z.shape
      len_z_shape = len(z_shape)
      if (
          (len_z_shape == 2 and z_shape[1] != 1)
          or (len_z_shape == 0 and not z.numel())
          or (len_z_shape > 2)
      ):
          raise ValueError(
              f"Invalid input size, we expect [B]. Got: {z.shape}"
          )
  
file_path: kornia/kornia/geometry/liegroup/_utils.py
imports: from kornia.core import Tensor
label_desc: Use image node_id 1 for calling a global function with example usage: check_so2_z_shape(z) without return types