Thanks for your great work and released code. <img width="602" alt="截屏2022-08-31 10 38 01" src="https://user-images.githubusercontent.com/54984530/187580351-c9c9c607-aabe-4c15-b1b6-40f0a9ea7198.png"> Should "q.view(N, C)" be revised to "t.view(N, C)" ?
Thanks for your great work and released code.

Should "q.view(N, C)" be revised to "t.view(N, C)" ?