code: def an_array():
      return np.random.randn(10, 3, 32, 32)
  
file_path: baal/tests/utils/test_array_utils.py
imports: import numpy
  import pytest
  import torch
  from scipy.special import softmax, expit
  from baal.utils import array_utils
  from baal.utils.array_utils import to_prob
  from baal.utils.iterutils import map_on_tensor
label_desc: Use image node_id 2 for calling a global function with example usage: an_array() and returns: np