https://catalogcasiokeyboardlk.blogspot.com/2023/10/reduce-array-of-matrices-in-tensorflow.html