https://mingredbestcheapprice.blogspot.com/2022/08/tensorflow-equivalent-for-numpy-indexed.html