在Tensorflow中循环参差不齐的张量

Question

我想知道是否有任何方法可以像tf.map_fn一样绕过参差不齐的张量。我的参差不齐的张量具有不同的行数，但包含4个要检索的点。

输入内容如下：

ragged_tensor[0] equals (100, 4)
ragged_tensor[1] equals (50, 4)

现在，我可以通过在RaggedTensor内部的第一个张量上循环来检索所有点：

test = tf.map_fn(lambda box: tf.image.crop_to_bounding_box(img, box[0], box[1], box[2], box[3]), tf.cast(boxes, tf.int32), dtype=tf.float32)

有人对此有任何经验吗，或者可能会给我一些提示和技巧？感谢所有帮助。

Answer 1

这是获取所有积分的一种方法：

points = tf.reshape(ragged_tensor.flat_values, [-1, 4])