W3cubDocs

/TensorFlow C++

tensorflow::ops::UnsortedSegmentProd

#include <math_ops.h>

Computes the product along segments of a tensor.

Summary

Read the section on segmentation for an explanation of segments.

This operator is similar to the unsorted segment sum operator found (here). Instead of computing the sum over segments, it computes the product of all entries belonging to a segment such that:

\(output_i = data_j\) where the product is over j such that segment_ids[j] == i.

If there is no entry for a given segment ID i, it outputs 1.

Arguments:

  • scope: A Scope object
  • segment_ids: A 1-D tensor whose rank is equal to the rank of data's first dimension.

Returns:

  • Output: Has same shape as data, except for dimension 0 which has size num_segments.
Constructors and Destructors
UnsortedSegmentProd(const ::tensorflow::Scope & scope, ::tensorflow::Input data, ::tensorflow::Input segment_ids, ::tensorflow::Input num_segments)
Public attributes
output
Public functions
node() const
::tensorflow::Node *
operator::tensorflow::Input() const
operator::tensorflow::Output() const

Public attributes

output

::tensorflow::Output output

Public functions

UnsortedSegmentProd

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

node

::tensorflow::Node * node() const 

operator::tensorflow::Input

operator::tensorflow::Input() const 

operator::tensorflow::Output

operator::tensorflow::Output() const 

© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/unsorted-segment-prod.html