dòng chảy :: hoạt động :: UnsortedSegmentProd
#include <math_ops.h>
Tính toán sản phẩm dọc theo các đoạn của 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.
Toán tử này tương tự như toán tử tổng phân đoạn không được sắp xếp được tìm thấy (ở đây) . Thay vì tính toán tổng trên các phân đoạn, nó tính tích của tất cả các mục nhập thuộc một phân đoạn sao cho:
\(output_i = {j...} data[j...]\) trong đó sản phẩm trên bộ j...
như vậy mà segment_ids[j...] == i
.
Ví dụ:
c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]]) tf.unsorted_segment_prod(c, tf.constant([0, 1, 0]), num_segments=2) # ==> [[ 4, 6, 6, 4], # [5, 6, 7, 8]]
Nếu không có mục nhập nào cho ID phân đoạn đã cho i
, nó sẽ xuất ra 1.
Nếu ID phân đoạn đã cho i
là số âm, thì giá trị tương ứng sẽ bị loại bỏ và sẽ không được đưa vào kết quả.
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ướcsegment_ids.rank
đầu tiên, được thay thế bằng một thứ nguyên duy nhất có kích thướcnum_segments
.
Người xây dựng và Người phá hủy | |
---|---|
UnsortedSegmentProd (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
UnsortedSegmentProd
UnsortedSegmentProd( 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