diff options
Diffstat (limited to 'tensorflow/core/api_def/base_api/api_def_UnsortedSegmentProd.pbtxt')
-rw-r--r-- | tensorflow/core/api_def/base_api/api_def_UnsortedSegmentProd.pbtxt | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/tensorflow/core/api_def/base_api/api_def_UnsortedSegmentProd.pbtxt b/tensorflow/core/api_def/base_api/api_def_UnsortedSegmentProd.pbtxt index efbc023705..9c8ea3b620 100644 --- a/tensorflow/core/api_def/base_api/api_def_UnsortedSegmentProd.pbtxt +++ b/tensorflow/core/api_def/base_api/api_def_UnsortedSegmentProd.pbtxt @@ -3,15 +3,15 @@ op { in_arg { name: "segment_ids" description: <<END -A 1-D tensor whose rank is equal to the rank of `data`'s -first dimension. +A tensor whose shape is a prefix of `data.shape`. END } out_arg { name: "output" description: <<END -Has same shape as data, except for dimension 0 which -has size `num_segments`. +Has same shape as data, except for the first `segment_ids.rank` +dimensions, which are replaced with a single dimension which has size +`num_segments`. END } summary: "Computes the product along segments of a tensor." @@ -25,9 +25,12 @@ This operator is similar to the unsorted segment sum operator found Instead of computing the sum over segments, it computes the product of all entries belonging to a segment such that: -\\(output_i = \prod_j data_j\\) where the product is over `j` such -that `segment_ids[j] == i`. +\\(output_i = \prod_{j...} data[j...]\\) where the product is over tuples +`j...` such that `segment_ids[j...] == i`. If there is no entry for a given segment ID `i`, it outputs 1. + +If the given segment ID `i` is negative, then the corresponding value is +dropped, and will not be included in the result. END } |