HSI_SSFTT icon indicating copy to clipboard operation
HSI_SSFTT copied to clipboard

Published 20 hours ago • verified publisher icon zgr6010

SSFTTnet.py下SSFTTnet类下的forward

Open renxiaosa00 opened this issue 2 years ago • 11 comments

SSFTTnet.py下SSFTTnet类下的forward： “wa = rearrange(self.token_wA, 'b h w -> b w h') # Transpose A = torch.einsum('bij,bjk->bik', x, wa)” wa是torch.Size([1, 64, 4]) x是torch.Size([64, 81, 64]) 这块bij bjk里面的b不一样

renxiaosa00 avatar Oct 26 '22 09:10 renxiaosa00