https://mingredbestcheapprice.blogspot.com/2024/02/numpy-broadcasting-with-3d-arrays.html