https://camera-optical-zoom.blogspot.com/2024/05/tensorflow-bincount-with-axis-option.html