Convolutional neural network (CNN) - Pytorch版