Function flatten_layer

tests/validation/reference/FlattenLayer.cpp:35–43

{
template <typename T>
SimpleTensor<T> flatten_layer(const SimpleTensor<T> &src, const TensorShape &shape_flatten)
{
    SimpleTensor<T> dst(shape_flatten, src.data_type(), 1);

    // Note: Since the reference implementation does not use padding bytes, we can copy directly the content of the source tensor
    std::copy(src.data(), src.data() + src.num_elements(), dst.data());

    return dst;
}

template SimpleTensor<float> flatten_layer(const SimpleTensor<float> &src, const TensorShape &shape_flatten);
template SimpleTensor<half>  flatten_layer(const SimpleTensor<half> &src, const TensorShape &shape_flatten);