在tf图中裁剪具有张量的张量

Question

我有一个张量，需要用张量的索引进行裁剪。

ex - Input (None,5x5x10)张量BoundingBox (None，2) --张量我希望有一个执行以下输出(None,3x2x10)的操作--如果边界框0，0= 3，边界框0，1=2，则张量

这与tf.image.crop_to_bounding_box相同，但此函数不会将张量类型的边界框作为输入。请帮帮忙。

Answer 1

不幸的是，对于“标准”张量运算，这是不可能的，因为输出的尺寸可能会发生变化。

考虑这样一个示例，其中bounding_box[0] == [3,2]和bounding_box[1] == [4,2]，那么您的输出形状需要为(None, 3 or 4, 2, 10)，并且(当然)具有维度3 or 4的标准张量是不允许的。

然而，TensorFlow确实有一个参差不齐的张量的概念，可以用来表示不同维度的作物，但这是一个不寻常的情况，不太可能适合大多数主流的下游训练操作。不过，阅读这篇文章可能是值得的，看看它是否适合您的用例：link