code: def test_basic_argmax_argmin(self):
      class SimpleOp(nn.Module):
          def forward(self, x):
              out1 = x.argmax()
              out2 = torch.argmax(x)
              out3 = x.argmax(dim=1)
              out4 = torch.argmax(x, dim=1)
              out5 = x.argmax(dim=1, keepdim=True)
              o1 = x.argmin()
              o2 = torch.argmin(x)
              o3 = x.argmin(dim=1)
              o4 = x.argmin(dim=1, keepdim=True)
              return out1, out2, out3, out4, out5, o1, o2, o3, o4
  
      self.checkExportImport(SimpleOp(), (torch.randn(4, 4),))
  
imports: import unittest
  import torch
  import nni.nas.nn.pytorch.layers
  from .convert_mixin import ConvertMixin, ConvertWithShapeMixin
label_desc: Use image node_id 16 for calling the TestConvert obj's underlying member method code with example usage: obj.test_basic_argmax_argmin() and returns: out1,  out2,  out3,  out4,  out5,  o1,  o2,  o3,  o4