code: def collect_stripes(matrix, stripes, group_width):
      subset = np.zeros((matrix.shape[0], len(stripes) * group_width))
      for s, stripe in enumerate(stripes):
          subset[
              ..., s * group_width : s * group_width + group_width
          ] = matrix[
              ...,
              stripe * group_width : stripe * group_width + group_width,
          ]
      return subset
  
file_path: apex/apex/contrib/sparsity/permutation_search_kernels/exhaustive_search.py
imports: from .permutation_utilities import *
  import pickle
  import os.path
  from os import path
  import math
label_desc: Use image node_id 6 for calling a global function with example usage: collect_stripes(matrix, stripes, group_width) and returns: subset