我在 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] 相对应的部分。
我怎样才能做到这一点？
最佳答案
您可以使用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]
关于python - 使用张量索引对张量进行切片，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/55359922/