Function sample_range

include/dali/core/tensor_shape.h:1489–1501

 */
template <int out_ndim, int ndim>
void sample_range(TensorListShape<out_ndim> &out, const TensorListShape<ndim> &in,
                  int begin, int end, int step = 1) {
  detail::check_compatible_ndim<out_ndim, ndim>();
  assert(begin >= 0 && begin <= in.num_samples());
  assert(end >= begin && end <= in.num_samples());

  int nsamples = div_ceil(end - begin, step);
  out.resize(nsamples, in.sample_dim());

  for (int i = begin, j = 0; j < nsamples; i += step, j++) {
    out.set_tensor_shape(j, in.tensor_shape_span(i));
  }
}

/**
 * Returns a (strided) range of sample shapes as a new TensorListShape.