pytorch定义神经卷积网络CNN源码

import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
# torch.nn仅支持迷你批次。整个torch.nn 软件包仅支持小批量样本的输入，而不是单个样本。
#
# 例如，nn.Conv2d将采用4D Tensor of 。nSamples x nChannels x Height x Width
#
# 如果您有一个样本，只需使用input.unsqueeze(0)添加假批量维度。

# 定义神经网络类
class