exemplar_att = exemplar_att.view(-1, self.channel, fea_size[0], fea_size[1]) # N,C1,H*W -> N,C1,H,W
exemplar_out = self.conv1(exemplar_att + exemplar)
query_att = query_att.view(-1, self.channel, fea_size[0], fea_size[1]) # N,C2,H*W -> N,C2,H,W
query_out = self.conv1(query_att + query)