https://keroseneheaterreview.blogspot.com/2024/06/tensorflow-reshaping-tensor.html