使用PyTorch实现卷积神经网络