如何将matlab代码“mat1（mat2（1:idx，2））=0”转换为torch？

如何将matlab代码mat1mat21:idx，2=0转换为无回路的torch？因为下面的循环版本确实需要时间。 这里mat1是一个长度为n的向量，mat2是一个matn，2

---

请报告一个输入输出真实测试，以便我们能够理解您想要实现什么。@AlessandroSuglia上下文有点复杂，因此我决定不在这里解释。

for i=1,idx do
    local idx = mat2[2][i]
    mat1[idx] = 0
end

mat1:indexFill(1,mat2[{2, {1,indx}}]:long(),0)