class_docstr: AnyNet stage (sequence of blocks w/ the same output shape).
class_docstr_tok: ["AnyNet","stage","(","sequence","of","blocks","w\/","the","same","output","shape",")","."]
code: def complexity(cx, w_in, w_out, stride, d, block_fun, params):
      for _ in range(d):
          cx = block_fun.complexity(cx, w_in, w_out, stride, params)
          stride, w_in = 1, w_out
      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, SE
  from torch.nn import Module
label_desc: Use image node_id 3 for calling the AnyStage obj's underlying member method code with example usage: obj.complexity(cx, w_in, w_out, stride, d, block_fun, params) and returns: cx