class_docstr: Stage of ResNet.
class_docstr_tok: ["Stage","of","ResNet","."]
code: def complexity(cx, w_in, w_out, stride, d, w_b=None, groups=1):
      for i in range(d):
          b_stride = stride if i == 0 else 1
          b_w_in = w_in if i == 0 else w_out
          trans_f = get_trans_fun(cfg.RESNET.TRANS_FUN)
          cx = ResBlock.complexity(
              cx, b_w_in, w_out, b_stride, trans_f, w_b, groups
          )
      return cx
  
imports: from pycls.core.config import cfg
  from pycls.models.blocks import activation, conv2d, conv2d_cx, gap2d, gap2d_cx, init_weights, linear, linear_cx, norm2d, norm2d_cx, pool2d, pool2d_cx
  from torch.nn import Module
label_desc: Use image node_id 3 for calling the ResStage obj's underlying member method code with example usage: obj.complexity(cx, w_in, w_out, stride, d, w_b, groups) and returns: cx