I’d like to concatenate each input image with 2d matrices but I’m stuck with errors.
I get (<class ‘RuntimeError’>, RuntimeError(‘Sizes of tensors must match except in dimension 0. Got 1 and 8 (The offending index is 0)’))
since the input_img has size of {Tensor: (8, 3, 256, 256)} and 2d matrices have size of {Tensor: (1, 1, 256, 256)}
input image size = 3x256x256
batch size = 8
Please help me out with this problem…! Any advice will be appreciated!