code: def create_dense_flows(
      flattened_flows, batch_size, image_height, image_width
  ):
      # possibly .view
      return torch.reshape(
          flattened_flows, [batch_size, image_height, image_width, 2]
      )
  
file_path: espnet/espnet/utils/spec_augment.py
imports: import random
  import torch
label_desc: Use image node_id 9 for calling a global function with example usage: create_dense_flows(flattened_flows, batch_size, image_height, image_width) and returns: torch