code: def gen_random_image(shape, rng):
      return torch.tensor(
          rng.uniform(size=shape), dtype=torch.get_default_dtype()
      )
  
file_path: NeuralCompression/tests/test_models.py
imports: import numpy
  import pytest
  import torch
  from utils import create_input
  import neuralcompression.functional
  from neuralcompression.models.deep_video_compression import DVC, DVCCompressionDecoder, DVCCompressionEncoder, DVCMotionCompensationModel, DVCPyramidFlowEstimator
label_desc: Use image node_id 1 for calling a global function with example usage: gen_random_image(shape, rng) and returns: torch