Open starryFelix opened 3 years ago
corr = corr.view(batch, ht, wd, 1, ht, wd) corr = corr / torch.sqrt(torch.tensor(dim).float()) corr = corr.view(-1, 1, ht, wd)
Thank you very much for your work , but I did not understand this part of the code.Can you tell me ? Thank you.
corr = corr.view(batch, ht, wd, 1, ht, wd) corr = corr / torch.sqrt(torch.tensor(dim).float()) corr = corr.view(-1, 1, ht, wd)
Thank you very much for your work , but I did not understand this part of the code.Can you tell me ? Thank you.