Function slice

tests/validation/reference/SliceOperations.cpp:38–73

{
template <typename T>
SimpleTensor<T> slice(const SimpleTensor<T> &src, Coordinates starts, Coordinates ends)
{
    using namespace arm_compute::helpers::tensor_transform;

    // Validation checks
    ARM_COMPUTE_ERROR_ON(src.shape().num_dimensions() > 4);
    ARM_COMPUTE_ERROR_ON(starts.num_dimensions() > src.shape().num_dimensions());
    ARM_COMPUTE_ERROR_ON(
        std::any_of(starts.cbegin(), starts.cbegin() + starts.num_dimensions(), [](int i) { return i < 0; }));
    ARM_COMPUTE_ERROR_ON(ends.num_dimensions() > src.shape().num_dimensions());

    // Get source shape
    const TensorShape &src_shape = src.shape();

    // Get destination shape
    TensorShape dst_shape = arm_compute::misc::shape_calculator::compute_slice_shape(src_shape, starts, ends);

    // Create destination tensor
    SimpleTensor<T> dst{dst_shape, src.data_type(), 1};

    // Perform slice
    Window win;
    win.use_tensor_dimensions(dst_shape);
    execute_window_loop(win,
                        [&](const Coordinates &id)
                        {
                            Coordinates offset;
                            for (unsigned int i = 0; i < id.num_dimensions(); ++i)
                            {
                                offset.set(i, starts[i] + id[i]);
                            }
                            *reinterpret_cast<T *>(dst(id)) = *reinterpret_cast<const T *>(src(offset));
                        });

    return dst;
}

template SimpleTensor<float> slice(const SimpleTensor<float> &src, Coordinates starts, Coordinates ends);
template SimpleTensor<half_float::half>