B， C ，W ，H …
关于torch.bmm()函数计算过程
batch size不动
torch.bmm(input, mat2, out=None) → Tensor
并且相乘的两个矩阵，要满足一定的维度要求：input（p,m,n) * mat2(p,n,a) ->output(p,m,a)。这个要求，可以类比于矩阵相乘。前一个矩阵的列等于后面矩阵的行才可以相乘。
torch.bmm() 与 torch.matmul()
pytorch之nn.Conv1d和nn.Conv2d超详解
nn.Conv1d 对C做通道变换，对W做卷积
nn.Conv2d 对C做通道变换，对W, H做卷积
parser.add_argument：action的作用
