Created by: qianledan
paddle的动图用topk在模型中,结果出现了显存溢出。实在没有办法重新学习pytorch,在pytorch中同样的模型结构用topk就能够正常跑程序。
另外:另外,paddle中的gather(input,index),index的rank=1,只能指定最外层的维度条目。跟pytorch中的ganther区别很大,功能已经被阉割了。希望能够将改函数的功能做成同pytorch中的一样。