Stay organized with collections
Save and categorize content based on your preferences.
tensorflow::
ops::
SegmentSum
#include <math_ops.h>
Computes the sum along segments of a tensor.
Summary
Read
the section on segmentation
for an explanation of segments.
Computes a tensor such that \(output_i = data_j\) where sum is over
j
such that
segment_ids[j] == i
.
If the sum is empty for a given segment ID
i
,
output[i] = 0
.
For example:
c = tf.constant([[1,2,3,4], [4, 3, 2, 1], [5,6,7,8]])
tf.segment_sum(c, tf.constant([0, 0, 1]))
# ==> [[5, 5, 5, 5],
# [5, 6, 7, 8]]
Args:
-
scope: A
Scope
object
-
segment_ids: A 1-D tensor whose size is equal to the size of
data
's first dimension. Values should be sorted and can be repeated.
Returns:
-
Output
: Has same shape as data, except for dimension 0 which has size
k
, the number of segments.
Public attributes
Public functions
node
::tensorflow::Node * node() const
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2021-05-14 UTC.
[null,null,["Last updated 2021-05-14 UTC."],[],[],null,["# tensorflow::ops::SegmentSum Class Reference\n\ntensorflow::\nops::\nSegmentSum\n=============================\n\n`\n#include \u003cmath_ops.h\u003e\n`\n\n\nComputes the sum along segments of a tensor.\n\nSummary\n-------\n\n\nRead\n[the section on segmentation](https://tensorflow.org/api_docs/python/tf/math#Segmentation)\nfor an explanation of segments.\n\n\nComputes a tensor such that \\\\(output_i = data_j\\\\) where sum is over\n`\nj\n`\nsuch that\n`\nsegment_ids[j] == i\n`\n.\n\n\nIf the sum is empty for a given segment ID\n`\ni\n`\n,\n`\noutput[i] = 0\n`\n.\n\n\n\u003cbr /\u003e\n\n\nFor example:\n\n\n```gdscript\nc = tf.constant([[1,2,3,4], [4, 3, 2, 1], [5,6,7,8]])\ntf.segment_sum(c, tf.constant([0, 0, 1]))\n# ==\u003e [[5, 5, 5, 5],\n# [5, 6, 7, 8]]\n```\n\n\u003cbr /\u003e\n\n\nArgs:\n\n- scope: A [Scope](/versions/r2.5/api_docs/cc/class/tensorflow/scope#classtensorflow_1_1_scope) object\n- segment_ids: A 1-D tensor whose size is equal to the size of `\n data\n ` 's first dimension. Values should be sorted and can be repeated.\n\n\u003cbr /\u003e\n\n\nReturns:\n\n- `\n `[Output](/versions/r2.5/api_docs/cc/class/tensorflow/output#classtensorflow_1_1_output)`\n ` : Has same shape as data, except for dimension 0 which has size `\n k\n ` , the number of segments.\n\n\u003cbr /\u003e\n\n| ### Constructors and Destructors ||\n|---|---|\n| ` `[SegmentSum](#classtensorflow_1_1ops_1_1_segment_sum_1a4f3c70df95d8be6a28060d01172358b5)` (const :: `[tensorflow::Scope](/versions/r2.5/api_docs/cc/class/tensorflow/scope#classtensorflow_1_1_scope)` & scope, :: `[tensorflow::Input](/versions/r2.5/api_docs/cc/class/tensorflow/input#classtensorflow_1_1_input)` data, :: `[tensorflow::Input](/versions/r2.5/api_docs/cc/class/tensorflow/input#classtensorflow_1_1_input)` segment_ids) ` ||\n\n| ### Public attributes ||\n|-----------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------|\n| ` `[operation](#classtensorflow_1_1ops_1_1_segment_sum_1a2a06a8d703c2aaac86320935d50be188)` ` | ` `[Operation](/versions/r2.5/api_docs/cc/class/tensorflow/operation#classtensorflow_1_1_operation)` ` |\n| ` `[output](#classtensorflow_1_1ops_1_1_segment_sum_1a475eb3ccf98c3aa9097ae79ad39b06ac)` ` | ` :: `[tensorflow::Output](/versions/r2.5/api_docs/cc/class/tensorflow/output#classtensorflow_1_1_output)` ` |\n\n| ### Public functions ||\n|---------------------------------------------------------------------------------------------------------------------------|--------------------------|\n| ` `[node](#classtensorflow_1_1ops_1_1_segment_sum_1a02ad69e7aabc1982f43a6f669d6dd884)` () const ` | ` ::tensorflow::Node * ` |\n| ` `[operator::tensorflow::Input](#classtensorflow_1_1ops_1_1_segment_sum_1ab999a101c40c98e2574092d55e877ec2)` () const ` | ` ` |\n| ` `[operator::tensorflow::Output](#classtensorflow_1_1ops_1_1_segment_sum_1af298372906f96c26410a5edaeb58a2e4)` () const ` | ` ` |\n\nPublic attributes\n-----------------\n\n### operation\n\n```text\nOperation operation\n``` \n\n### output\n\n```text\n::tensorflow::Output output\n``` \n\nPublic functions\n----------------\n\n### SegmentSum\n\n```gdscript\n SegmentSum(\n const ::tensorflow::Scope & scope,\n ::tensorflow::Input data,\n ::tensorflow::Input segment_ids\n)\n``` \n\n### node\n\n```gdscript\n::tensorflow::Node * node() const \n``` \n\n### operator::tensorflow::Input\n\n```gdscript\n operator::tensorflow::Input() const \n``` \n\n### operator::tensorflow::Output\n\n```gdscript\n operator::tensorflow::Output() const \n```"]]
