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