code: def _check_input_dim(self, xr, xi):
      assert xr.shape == xi.shape
      assert xr.size(1) == self.num_features
  
imports: import torch
  import torch.nn
  import torch.nn.functional
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