用户
 找回密码
 立即注册
发表于 2021-3-5 15:15:05
67830
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]]


使用道具 举报 回复
发新帖
您需要登录后才可以回帖 登录 | 立即注册