小编use*_*523的帖子

假设我有一个类似[2,3,4]的数组，我正在NumPy（或Tensorflow）中寻找一种将其转换为[0,0,1,1,1,2,2,2,2,2]的方法将tf.math.segment_sum（）应用于大小为2 + 3 + 4的张量。

我脑海中没有一个绝妙的主意，只有循环和列表理解。