code: def make_5d_fake_dist(means, stds, dims=10):
      d = np.stack(
          [
              make_3d_fake_dist(means, stds, dims=dims)
              for _ in range(IMG_SIZE**2)
          ],
          -1,
      )  # 3x3 image
      b, c, i, hw = d.shape
      d = np.reshape(d, [b, c, i, IMG_SIZE, IMG_SIZE])
      d = np.rollaxis(d, 2, 5)
      # [n_sample, n_class, H, W, iter]
      return d
  
imports: import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: make_5d_fake_dist(means, stds, dims) and returns: d