code: def arange_4d_image(request):
      x = torch.arange(torch.prod(torch.tensor(request.param))).reshape(
          request.param
      )
  
      return x.to(torch.get_default_dtype())
  
file_path: NeuralCompression/tests/conftest.py
imports: import pytest
  import torch
  import torch.nn
  from torch import Tensor
label_desc: Use image node_id 1 for calling a global function with example usage: arange_4d_image(request) and returns: x