https://camera-optical-zoom.blogspot.com/2024/02/tensorflow-tfgather-with-axis-parameter.html