code: def _check_input_dim(self, xr, xi):
      assert xr.shape == xi.shape
      assert xr.size(1) == self.num_features
  
imports: import functools
  from functools import partial
  import numpy
  import torch
  from torch import Tensor, nn
  from torch.nn.modules.batchnorm import _BatchNorm
label_desc: Use image node_id 4 for calling the ComplexBatchNorm obj's underlying member method code with example usage: obj._check_input_dim(xr, xi) without return types