https://camera-optical-zoom.blogspot.com/2024/03/tensorflow-equivalent-for-numpy-indexed.html