code: def init(self, M, N, chunks, device):
      self.inputs = {
          "input_one": torch.rand(M, N, device=device),
          "chunks": chunks,
      }
      self.set_module_name("chunk")
  
imports: import torch
  import operator_benchmark
label_desc: Use image node_id 1 for calling the ChunkBenchmark obj's underlying member method code with example usage: obj.init(M, N, chunks, device) without return types