code: def choice_blocks(layers_configs, stage_ops):
      """
      Create list of layer candidates for NNI one-shot NAS.
  
      Parameters
      ----------
      layers_configs : list
          the layer config: [input_channel, output_channel, stride, height]
      stage_ops : dict
          the pairs of op name and layer operator
  
      Returns
      -------
      output: list
          list of layer operators
      """
      ops_names = [op for op in stage_ops]
      fm = {"fm_size": layers_configs[3]}
      op_list = [
          stage_ops[op](*layers_configs[0:3], **fm) for op in ops_names
      ]
  
      return op_list
  
imports: from __future__ import absolute_import, division, print_function
  import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 2 for calling a global function with example usage: choice_blocks(layers_configs, stage_ops) and returns: op_list