我正在研究在Tensorflow中实现3d检测架构Lasernet。我坚持的步骤涉及从范围图像中进行每点边界框预测，将其投影到xy平面上，并执行聚类。这些预测将映射到0.5mx0.5m的离散单元，并且多个预测可以落入每个单元。然后，基于加权均值合并单元。我遇到的困难是如何将预测存储在2d网格上。最方便的方法是让每个网格单元都有一个索引列表，这些索引对应于其内的范围图像点。唯一的问题是，这引入了可变大小的最内维度，因为任意数量的点都可以属于单个像元。是否可以在tensorflow中表示此映射，或者我需要创建自定义内核来实现这一点？