code: def forward(self, input_one, chunks: int):
      return torch.chunk(input_one, chunks)
  
imports: import torch
  import operator_benchmark
label_desc: Use image node_id 2 for calling the ChunkBenchmark obj's underlying member method code with example usage: obj.forward(input_one, chunks) and returns: torch