python - 使用张量索引对张量进行切片

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

您可以使用tf.Tensor.__getitem__与 NumPy 索引非常相似:

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

或者，使用 tf.slice :

sliced_img = tf.slice(img, [0, h_start, w_start], [-1, h_end - h_start, w_end-w_start]