for making mask i used from this code but have error:
mask= torch.cat([torch.ones([8,8,2,3]), torch.zeros([8,8,1,3])],0)
error: RuntimeError: Sizes of tensors must match except in dimension 0. Got 2 and 1 in dimension 2 (The offending index is 1)
dimension of output activation of self.conv3 is [8,8,3,3] and i would to repeat end row and column to one before the last row and column.
thanks for you guidance…