https://blog.ando.ai/posts/tensor-puzzles/