เทนเซอร์โฟลว์:: ปฏิบัติการ:: ผลรวมกลุ่ม
#include <math_ops.h>
คำนวณผลรวมตามส่วนของเทนเซอร์
สรุป
อ่าน ส่วนเกี่ยวกับการแบ่งส่วน เพื่อดูคำอธิบายของส่วนต่างๆ
คำนวณเทนเซอร์เช่นนั้น \(output_i = data_j\) โดยที่ผลรวมมากกว่า j
เช่นว่า segment_ids[j] == i
หากผลรวมว่างเปล่าสำหรับ ID เซ็กเมนต์ที่กำหนด i
output[i] = 0
ตัวอย่างเช่น:
c = tf.constant([[1,2,3,4], [4, 3, 2, 1], [5,6,7,8]]) tf.segment_sum(c, tf.constant([0, 0, 1])) # ==> [[5, 5, 5, 5], # [5, 6, 7, 8]]
ข้อโต้แย้ง:
- ขอบเขต: วัตถุ ขอบเขต
- Segment_ids: เทนเซอร์ 1 มิติที่มีขนาดเท่ากับขนาดของมิติแรกของ
data
ควรเรียงลำดับค่าและสามารถทำซ้ำได้
ผลตอบแทน:
-
Output
: มีรูปร่างเหมือนกับข้อมูล ยกเว้นมิติ 0 ซึ่งมีขนาดk
คือจำนวนเซ็กเมนต์
ตัวสร้างและผู้ทำลาย | |
---|---|
SegmentSum (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input segment_ids) |
คุณลักษณะสาธารณะ | |
---|---|
operation | |
output |
งานสาธารณะ | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
คุณลักษณะสาธารณะ
การดำเนินการ
Operation operation
เอาท์พุท
::tensorflow::Output output
งานสาธารณะ
ผลรวมกลุ่ม
SegmentSum( const ::tensorflow::Scope & scope, ::tensorflow::Input data, ::tensorflow::Input segment_ids )
โหนด
::tensorflow::Node * node() const
ตัวดำเนินการ::tensorflow::อินพุต
operator::tensorflow::Input() const
ตัวดำเนินการ::tensorflow::เอาต์พุต
operator::tensorflow::Output() const