Блог

Я использую сегмент tf. Входные данные-это два списка (преобразуются в тензор)

 def run():
    res = {}
    data = [11,22,33,44,55]
    segment_ids = [6,6,7,7,8]
    data = tf.convert_to_tensor(data)
    segment_ids = tf.convert_to_tensor(segment_ids)
    # some operation
    # ...
    return res
 

для вывода я хочу получить следующее:

 res[0] = tensor (value: [11,22])
res[1] = tensor (value: [33,44])
res[2] = tensor (value: [55])
 

ключом res является индекс различных идентификаторов сегментов ( индекс списка(6,7,8): [0,1,2] ).
значение — это соответствующие данные идентификаторов каждого сегмента.

соответствует нулевому индексу, тогда в этом случае значение res[0] равно [11,11]