Pytorch Print Layers For Backwar