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