Có một câu hỏi? Kết nối với cộng đồng tại Diễn đàn Truy cập Diễn đàn TensorFlow

dòng chảy :: hoạt động :: UnsortedSegmentSum

#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] = {j...} data[j...]\) trong đó tổng trên các bộ giá trị j... sao cho segment_ids[j...] == i . Không giống như SegmentSum , segment_ids không cần được sắp xếp và không cần bao hàm tất cả các giá trị trong toàn bộ dải giá trị hợp lệ.

Nếu tổng trống cho một ID phân đoạn nhất định i , output[i] = 0 . Nếu ID phân đoạn đã cho i là số âm, giá trị sẽ bị giảm và sẽ không được thêm vào tổng của phân đoạn.

num_segments phải bằng số lượng ID phân đoạn riêng biệt.

c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]])
tf.unsorted_segment_sum(c, tf.constant([0, 1, 0]), num_segments=2)
# ==> [[ 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 có hình dạng là tiền tố của data.shape .

Lợi nhuận:

  • Output : Có cùng hình dạng với dữ liệu, ngoại trừ kích thước segment_ids.rank đầu tiên, được thay thế bằng một thứ nguyên duy nhất có kích thước num_segments .

Người xây dựng và Người phá hủy

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

Thuộc tính công khai

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 khai

hoạt động

Operation operation

đầu ra

::tensorflow::Output output

Chức năng công cộng

UnsortedSegmentSum

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

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