code: def _check_input_dim(self, input):
      if input.dim() < 2 or input.dim() > 4:
          raise ValueError(
              "expected 4D or 3D input (got {}D input)".format(
                  input.dim()
              )
          )
  
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 1 for calling the BatchNorm obj's underlying member method code with example usage: obj._check_input_dim(input) without return types