ti.kernel 中如何for遍历 ndarray类型的向量

如题:
self.grid_id_list是一个行数不固定,列数始终为2的2维数组,也就是里面都是长度为2的向量。
我想在@ti.kernel 里用for循环遍历,获取self.grid_id_list中每个元素的值:
for a in self.grid_id_list:
i = a[0]
j = a[1]

但是这样在@ti.kernel 里是行不通吗,怎么样才能实现这个功能呢。