I think you are again hitting the issue of trying to concatenate two tensors with a different shape as described e.g. in this post. This time you are trying to concatenate x and c in dim1 while dim0 shows the shape mismatch.