class_docstr: The full search space proposed by `NAS-Bench-201 <https://arxiv.org/abs/2001.00326>`__.
  
  It's a stack of :class:`~nni.nas.hub.pytorch.modules.NasBench201Cell`.
  
  Parameters
  ----------
  stem_out_channels
      The output channels of the stem.
  num_modules_per_stack
      The number of modules (cells) in each stack. Each cell is a :class:`~nni.nas.hub.pytorch.modules.NasBench201Cell`.
  num_labels
      Number of categories for classification.
class_docstr_tok: ["The","full","search","space","proposed","by","`","NAS-Bench-201","<","https",":","\/\/arxiv.org\/abs\/2001.00326",">","`","__",".","It","'s","a","stack","of",":","class",":","`","~nni.nas.hub.pytorch.modules.NasBench201Cell","`",".","Parameters","--","--","--","--","--","stem_out_channels","The","output","channels","of","the","stem",".","num_modules_per_stack","The","number","of","modules","(","cells",")","in","each","stack",".","Each","cell","is","a",":","class",":","`","~nni.nas.hub.pytorch.modules.NasBench201Cell","`",".","num_labels","Number","of","categories","for","classification","."]
code: def _make_op_factory(self, prim):
      return lambda C_in, C_out: OPS_WITH_STRIDE[prim](C_in, C_out, 1)
  
imports: from typing import Callable, Dict
  import torch
  import torch.nn
  from nni.nas.nn.pytorch import ModelSpace
  from .modules.nasbench201 import NasBench201Cell
label_desc: Use image node_id 2 for calling the NasBench201 obj's underlying member method code with example usage: obj._make_op_factory(prim) and returns: function