https://mingredbestcheapprice.blogspot.com/2024/01/tensorflow-bincount-with-axis-option.html