How to reshape Tensor properly?

Thank you. Based on your comment, I will try adjusting my tensor operation.