https://keroseneheaterreview.blogspot.com/2024/05/proper-way-to-define-inputshape-on.html