Function unstack

tests/validation/reference/Unstack.cpp:90–103

template <typename T>
std::vector<SimpleTensor<T>>
unstack(const SimpleTensor<T> &input_tensor, std::vector<SimpleTensor<T>> &output_tensors, int axis)
{
    // Wrap around negative values
    const unsigned int axis_u = wrap_around(axis, static_cast<int>(input_tensor.shape().num_dimensions()));
    ARM_COMPUTE_ERROR_ON(axis_u >= input_tensor.shape().num_dimensions());
    for (size_t k = 0; k < output_tensors.size(); ++k)
    {
        SimpleTensor<T>      &output    = output_tensors[k];
        const SimpleTensor<T> kth_slice = get_slice(input_tensor, axis_u, k);
        output                          = copy_tensor<T>(kth_slice);
    }
    return output_tensors;
}

template std::vector<SimpleTensor<float>>
unstack(const SimpleTensor<float> &input_tensor, std::vector<SimpleTensor<float>> &output_tensors, int axis);