code: def create_chunks(x, chunk_size=16384, chunk_stride=16384):
      """Splits the input into smaller chunks of size chunk_size with
      an overlap chunk_stride. The chunks are concatenated over
      the batch axis."""
      x = x.unfold(1, chunk_size, chunk_stride)
      x = x.reshape(x.shape[0] * x.shape[1], -1, 1)
      return x
  
imports: import os
  import sys
  import torch
  import torchaudio
  import speechbrain
  from pesq import pesq
  from hyperpyyaml import load_hyperpyyaml
  from speechbrain.utils.metric_stats import MetricStats
  from speechbrain.nnet.loss.stoi_loss import stoi_loss
  from speechbrain.utils.distributed import run_on_main
label_desc: Use image node_id 1 for calling a global function with example usage: create_chunks(x, chunk_size, chunk_stride) and returns: x