code: def fuse_bn(conv_or_fc: nn.Module, bn: nn.Module):
      std = (bn.running_var + bn.eps).sqrt()
      t = bn.weight / std
      t = t.reshape(-1, 1, 1, 1)
  
      if len(t) == conv_or_fc.weight.size(0):
          return (
              conv_or_fc.weight * t,
              bn.bias - bn.running_mean * bn.weight / std,
          )
      else:
          repeat_times = conv_or_fc.weight.size(0) // len(t)
          repeated = t.repeat_interleave(repeat_times, 0)
          return conv_or_fc.weight * repeated, (
              bn.bias - bn.running_mean * bn.weight / std
          ).repeat_interleave(repeat_times, 0)
  
label_desc: Use image node_id 1 for calling a global function with example usage: fuse_bn(conv_or_fc, bn) and returns: unknown,  unknown, unknown,  unknown