dòng chảy :: ops :: SegmentSum
#include <math_ops.h>
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ướck
, số lượng đoạn.
Người xây dựng và Người phá hủy | |
---|---|
SegmentSum (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input segment_ids) |
Thuộc tính công cộng | |
---|---|
operation | |
output |
Chức năng công cộng | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Thuộc tính công cộng
hoạt động
Operation operation
đầu ra
::tensorflow::Output output
Chức năng công cộng
SegmentSum
SegmentSum( const ::tensorflow::Scope & scope, ::tensorflow::Input data, ::tensorflow::Input segment_ids )
nút
::tensorflow::Node * node() const
toán tử :: tensorflow :: Đầu vào
operator::tensorflow::Input() const
toán tử :: tensorflow :: Đầu ra
operator::tensorflow::Output() const