Process input image through convolutions however you like, after last convolution insert nn.Flatten (or use any other method of converting tensors from shape (batch,channels,height,width) to shape (batch,data)), then torch.cat it with a features vector and pass it into final linear layer, or a couple of them