Function expand_coordinates

tests/validation/reference/Unstack.cpp:38–62

namespace
{
inline Coordinates expand_coordinates(Coordinates in_coord, size_t axis, size_t slice, size_t num_dimensions)
{
    /*
        Reconstruct input_coord to read the corresponding value from the correct slice. This is done by adding an extra dimension
        to the coordinates and shuffling around the values based on the info below.

        For example, if input tensor shape is (X, Y, Z, W);

        If axis == 0, each slice will have the shape (Y, Z, W) and there will be X slices

        If axis == 1, each slice will have the shape (X, Z, W) and there will be Y slices.
    */
    Coordinates expanded_coord;
    expanded_coord.set_num_dimensions(num_dimensions);
    expanded_coord.set(axis, slice);
    for (size_t k = 0; k < axis; ++k)
    {
        expanded_coord.set(k, in_coord[k]);
    }
    for (size_t k = axis + 1; k < num_dimensions; ++k)
    {
        expanded_coord.set(k, in_coord[k - 1]);
    }
    return expanded_coord;
}

template <typename T>
SimpleTensor<T> get_slice(const SimpleTensor<T> &input_tensor, size_t axis, size_t slice)