作者：東東1959 | 来源：互联网 | 2023-09-04 15:52
代码如下：
import numpy as np
import tensorflow as tf
gt_boxes = tf.constant(np.random.rand(2,10,4))
gt_counts = tf.constant([3,5])
boxes1 = tf.slice(gt_boxes,[0,0],tf.constant([1,3,4])) # --> 1x3x4
boxes2 = tf.slice(gt_boxes,[1,5,4])) # --> 1x5x4
有没有办法将gt_boxes和gt_counts结合在一起而没有索引？或者它需要tf.while_loop来处理这种索引。你能给点建议吗？谢谢。
京公网安备 11010802041100号