如题,我现在已经在一个函数A中,初始化了一个稀疏矩阵的builder,假设为A_Builder中的所有triplets,我现在需要在另一个函数B中去修改该稀疏矩阵中原有的部分值。
我想了两个方案:
- 将A中的A_builder给Build出来,获得稀疏矩阵A,然后作为参数,传入到函数B中,可是我尝试了一下是用A: ti.template(),结果失败了,我并不能在该函数中通过下标A[i, j]正常访问其中的值
- 遍历当前builder中已存在的triplets,并直接修改,可是我不知道该如何遍历triplets(print_triplets不算,因为我无法修改它)
不知道大家有没有什么好的方法,恳请大家指教!