https://discountedgerberprotectplus.blogspot.com/2022/06/numpy-broadcast-addition-along.html
Numpy Broadcast Addition Along Arbitrary Axes