https://keroseneheaterreview.blogspot.com/2024/10/apply-function-along-axis-over-two.html