tf.segment_sum和tf.unsorted_segment_sum

最新推荐文章于 2025-03-10 19:12:05 发布

原创

最新推荐文章于 2025-03-10 19:12:05 发布 · 698 阅读

1 ·

CC 4.0 BY-SA版权

本文介绍了 TensorFlow 中的 `tf.segment_sum` 和 `tf.unsorted_segment_sum` 函数，通过实例展示了它们的区别。`tf.segment_sum` 要求 segment_ids 有序，而 `tf.unsorted_segment_sum` 允许 segment_ids 无序，但输出仍按顺序排列。同时，`tf.unsorted_segment_sum` 还接受一个额外参数 num_segments，用于指定输出的段数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

c = tf.constant([[1,2,3,4], [-1,-2,-3,-4], [5,6,7,8]])
result = tf.segment_sum(c, tf.constant([0, 0, 1]))#第二个参数长度必须为3
result_ = tf.segment_sum(c, tf.constant([0, 1, 1]))
result__ = tf.segment_sum(c, tf.constant([0, 1, 2]))
result2 = tf.unsorted_segment_sum(c, tf.constant([2, 1, 1]),3)#第二个参数长度必须为3
result3 = tf.unsorted_segment_sum(c, tf.constant([1, 0, 1]),2)
#result4 = tf.unsorted_segment_sum(c, tf.constant([2, 0, 1]),2) #错误，segment_ids[0] = 2 is out of range [0, 2)
result4 = tf.unsorted_segment_sum(c, tf.constant([2, 0, 1]),3)
result5 = tf.unsorted_segment_sum(c, tf.constant([3, 1, 0]),5)
sess = tf.Session()
print("result")
print(sess.run(result))
print("result_")
print(sess.run(result_))
print("result__")
print(sess.run(result__))
print("result2")
print(sess.run(result2))
print("result3")
print(sess.run(result3))
print("result4")
print(sess.run(result4))
print("result5")
print(sess.run(result5))

输出结果：

result