【科研】代码总结

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的作用