code: def profiler():
      net = Net()
      profiler = FlopsProfiler(net, torch.randn(1, 1, 10, 10))
      # (9k, 25k, 49k)
      return profiler
  
file_path: nni/test/algo/nas/oneshot/test_profiler.py
imports: import pytest
  import torch
  from torch.nn import Conv2d
  from nni.nas.nn import ModelSpace, LayerChoice
  from nni.nas.oneshot.pytorch.profiler import RangeProfilerFilter, ExpectationProfilerPenalty, SampleProfilerPenalty
  from nni.nas.profiler.pytorch.flops import FlopsProfiler
label_desc: Use image node_id 1 for calling a global function with example usage: profiler() and returns: profiler