https://purchaseironmanelliptical.blogspot.com/2023/10/tensorflow-equivalent-for-numpy-indexed.html