torch.cat( (torch.max(x, 1)[0].unsqueeze(1), torch.mean(x, 1).unsqueeze(1)), dim=1 )

[Niki173]

triplet-attention.py

Hi,Run this file and find inconsistent dimensions

I wonder if there is a problem here

torch.cat( (torch.max(x, 1)[0].unsqueeze(1), torch.mean(x, 1).unsqueeze(1)), dim=1 )

it should be torch.cat( (torch.max(x, 1).unsqueeze(1), torch.mean(x, 1).unsqueeze(1)), dim=1 )

right?

[digantamisra98]

@Niki173 apologies for the delay in response, for Z-pool, we follow the same code as that given in CBAM