def forward(self, input, hidden): hx, cx = hidden **gates = self.conv_ih(input) + self.conv_hh(hx)** RuntimeError: The size of tensor a (32) must match the size of tensor b (18) at non-singleton dimension 0 i am using pytorch 0.3.1 version
def forward(self, input, hidden):
hx, cx = hidden
gates = self.conv_ih(input) + self.conv_hh(hx)
RuntimeError: The size of tensor a (32) must match the size of tensor b (18) at non-singleton dimension 0
i am using pytorch 0.3.1 version