https://capressocoffemakerthecheapest.blogspot.com/2024/01/numpy-broadcasting-with-3d-arrays.html