https://travelebookdownload.blogspot.com/2023/04/tensorflow-equivalent-for-numpy-indexed.html