code: def init(self, M, N, K, device, dtype):
      self.input_one = torch.rand(
          M, N, K, device=device, dtype=dtype, requires_grad=True
      )
      self.input_two = torch.rand(M, N, K, device=device, dtype=dtype)
      self.set_module_name("add")
  
imports: import operator_benchmark
  import torch
label_desc: Use image node_id 1 for calling the AddBenchmark obj's underlying member method code with example usage: obj.init(M, N, K, device, dtype) without return types