tensorflow - 在 Tensorflow 中循环不规则张量

Question

我想知道与 tf.map_fn 类似的 Ragged Tensors 是否有任何循环。我的 Ragged Tensor 有不同数量的行，但包含 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])