https://catalogcasiokeyboardlk.blogspot.com/2024/05/reshape-tensor-in-custom-loss-function.html