A comment would be nice to explain the final shape `(1, 600, 500, 2)` and its significance. For example, the lengths of axes 0 and 3 are arbitrary.