pytorch:定义自己的网络结构