code: def init(self, M, N, K, trans_a, trans_b, device):
      self.inputs = {
          "input_one": torch.rand(M, N, device=device)
          if trans_a
          else torch.rand(N, M, device=device).t(),
          "input_two": torch.rand(N, K, device=device)
          if trans_b
          else torch.rand(K, N, device=device).t(),
      }
      self.set_module_name("matmul")
  
imports: import torch
  import operator_benchmark
label_desc: Use image node_id 1 for calling the MatMulBenchmark obj's underlying member method code with example usage: obj.init(M, N, K, trans_a, trans_b, device) without return types