Если это растр, самое очевидное — последовательно соединить все строки или все столбцы входного изображения. Метод
flatten в Numpy. Если это полносвязная сеть (Dense-слои), не должно возникнуть никаких проблем.
Если это векторная фигура, например,
polyline, можно подать список координатов вершин.