看FFA原作者的代码学到的

for step in range(start_step+1,opt.steps+1):

为什么要把numpy转成tensor
“因为tensor是专门为GPU加速设计的矩阵,而numpy却不行。其实就相当于一个为GPU设计的数据结构。
添加链接描述

标签：学到,原作者,start,step,GPU,numpy,FFA,tensor
来源： https://blog.csdn.net/weixin_45883942/article/details/122141039