code: def test_function(self):
      return self.do_test(F.softmax, {"dim": 1})
  
file_path: nni/test/algo/compression/pruning/test_speedup_softmax.py
imports: import unittest
  import torch
  import torch.nn.functional
  from nni.compression.utils.counter import count_flops_params
  from nni.compression.speedup import ModelSpeedup
  from nni.compression.pruning import L1NormPruner
label_desc: Use image node_id 2 for calling the SpeedupSoftmaxTestCase obj's underlying member method code with example usage: obj.test_function() and returns: self