https://camera-optical-zoom.blogspot.com/2024/05/tensorflow-indexing-with-boolean-tensor.html