如何在2D矩阵中找到非零子组?

我目前正在使用Luajit Torch工作,并遇到了一个问题。 我有一个Tensor(2D矩阵),其数字可以是任何正整数或零。我需要将所有不为零的值分组,并找到每个组的质心。我正在尝试将问题解决为灰度图像(其中0将是黑色区域),首先尝试查找非黑色区域,然后计算这些区域的质心,但我尚未找到算法。

我仍在努力找到子组的索引,谢谢!

点赞