class_docstr: Test model to wrap around a single op for export.
class_docstr_tok: ["Test","model","to","wrap","around","a","single","op","for","export","."]
code: def __init__(self, op, kwargs):
      super().__init__()
      self.operator = op
      self.kwargs = kwargs
  
imports: from __future__ import annotations
  import copy
  import itertools
  from typing import Any, Callable, Collection, Mapping, Optional, Tuple, Type, Union
  import onnx_test_common
  import parameterized
  import pytest
  import pytorch_test_common
  import torch
  from onnx_test_common import skip, skip_slow, xfail
  from torch.onnx._internal.diagnostics import _rules
  from torch.testing._internal import common_device_type, common_methods_invocations, common_utils
  from torch.testing._internal.opinfo import core
label_desc: Use image node_id 1 to create a new SingleOpModel object from inherited base classes: torch.nn with example: obj = SingleOpModel(op, kwargs)