dòng chảy :: ops :: SegmentSum

Tính tổng dọc theo các đoạn của một tensor.

Tóm lược

Đọc phần về phân đoạn để biết giải thích về các phân đoạn.

Tính toán một tensor sao cho \(output_i = data_j\) trong đó tổng trên j sao cho segment_ids[j] == i .

Nếu tổng trống cho một ID phân đoạn cho trước i , output[i] = 0 .

Ví dụ:

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]]

Tranh luận:

• phạm vi: Một đối tượng Phạm vi
• segment_ids: Một tensor 1-D có kích thước bằng kích thước của kích thước đầu tiên của data . Các giá trị nên được sắp xếp và có thể được lặp lại.

Lợi nhuận:

• Output : Có hình dạng giống như dữ liệu, ngoại trừ kích thước 0 có kích thước k , số lượng đoạn.

Thuộc tính công cộng

Operation operation

::tensorflow::Output output

 SegmentSum(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input data,
  ::tensorflow::Input segment_ids
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const