cuda核函数小白,这种循环怎么写核函数
注:a,b,c为tensor,其中c中的值为0到31654。
a = [256,3,3,31654]b = [256,135,240]
c = [135,240]
for i in range(135):
for j in range(240):
b[:,i:3,j:3] += a[:,3,3,c[i,j]]
cuda核函数小白,这种循环怎么写核函数
注:a,b,c为tensor,其中c中的值为0到31654。
a = [256,3,3,31654]b = [256,135,240]
c = [135,240]
for i in range(135):
for j in range(240):
b[:,i:3,j:3] += a[:,3,3,c[i,j]]