code: def test_stack_in_memory_single(a_tensor):
      iterations = 10
      out = array_utils.stack_in_memory(a_tensor, iterations=iterations)
      assert out.shape == (10 * iterations, 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 4 for calling a global function with example usage: test_stack_in_memory_single(a_tensor) without return types