In this publication titled print shape of pytorch tensor, I have provided a complete image from start to finish ,Please look below
