dòng chảy :: ops :: SparseSegmentSum
#include <math_ops.h>
Tính tổng dọc theo các đoạn thưa thớt 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.
Giống như SegmentSum
, nhưng segment_ids
có thể có xếp hạng thấp hơn thứ nguyên đầu tiên của data
, chọn một tập hợp con của thứ nguyên 0, được chỉ định bởi indices
.
Ví dụ:
c = tf.constant([[1,2,3,4], [-1,-2,-3,-4], [5,6,7,8]])
# Select two rows, one segment. tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 0])) # => [[0 0 0 0]]
# Select two rows, two segment. tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 1])) # => [[ 1 2 3 4] # [-1 -2 -3 -4]]
# Select all rows, two segments. tf.sparse_segment_sum(c, tf.constant([0, 1, 2]), tf.constant([0, 0, 1])) # => [[0 0 0 0] # [5 6 7 8]]
# Which is equivalent to: tf.segment_sum(c, tf.constant([0, 0, 1]))
Tranh luận:
- phạm vi: Một đối tượng Phạm vi
- chỉ số: Một tensor 1-D. Có cùng thứ hạng với giá
segment_ids
. - segment_ids: Một tensor 1-D. 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 | |
---|---|
SparseSegmentSum (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input indices, :: 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
SparseSegmentSum
SparseSegmentSum( const ::tensorflow::Scope & scope, ::tensorflow::Input data, ::tensorflow::Input indices, ::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