https://devhubby.com/thread/how-to-print-tensor-shape-in-tensorflow