使用张量索引切片张量

Question

我在TF中有一个张量img，代表一张图像，其形状为(n_channels, img_height, img_width)。

我还有两个整数张量，h_start，h_end，w_start，w_end。

我想提取图像中与numpy中的img[:, :, h_start:h_end, w_start:w_end]相对应的部分。

我该怎么做？

Answer 1

您可以像使用NumPy索引一样使用docs：

UNIFIED_AUDIT_TRAIL

或者，使用tf.Tensor.__getitem__：

img[:, h_start:h_end, w_start:w_end]