pytorch搭建网络的四种写法