tf.math.unsorted_segment_mean
Stay organized with collections
Save and categorize content based on your preferences.
Computes the mean along segments of a tensor.
tf.math.unsorted_segment_mean(
data, segment_ids, num_segments, name=None
)
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 mean of all
entries belonging to a segment such that:
\(output_i = 1/N_i \sum_{j...} data[j...]\) where the sum is over tuples
j... such that segment_ids[j...] == i with \N_i\ being the number of
occurrences of id \i\.
If there is no entry for a given segment ID i, it outputs 0.
If the given segment ID i is negative, the value is dropped and will not
be added to the sum of the segment.
Args |
|---|
data
|
A Tensor with floating point or complex dtype.
|
segment_ids
|
An integer tensor whose shape is a prefix of data.shape.
|
num_segments
|
An integer scalar Tensor. The number of distinct segment
IDs.
|
name
|
A name for the operation (optional).
|
Returns |
|---|
A Tensor. 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.
|
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.
Last updated 2020-10-01 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2020-10-01 UTC."],[],[],null,["# tf.math.unsorted_segment_mean\n\n|---------------------------------------------------------------------------------------|-----------------------------------------------------------------------------------------------------------------------------|\n| [TensorFlow 1 version](/versions/r1.15/api_docs/python/tf/math/unsorted_segment_mean) | [View source on GitHub](https://github.com/tensorflow/tensorflow/blob/v2.0.0/tensorflow/python/ops/math_ops.py#L3527-L3570) |\n\nComputes the mean along segments of a tensor.\n\n#### View aliases\n\n\n**Compat aliases for migration**\n\nSee\n[Migration guide](https://www.tensorflow.org/guide/migrate) for\nmore details.\n\n[`tf.compat.v1.math.unsorted_segment_mean`](/api_docs/python/tf/math/unsorted_segment_mean), [`tf.compat.v1.unsorted_segment_mean`](/api_docs/python/tf/math/unsorted_segment_mean)\n\n\u003cbr /\u003e\n\n tf.math.unsorted_segment_mean(\n data, segment_ids, num_segments, name=None\n )\n\nRead [the section on\nsegmentation](https://tensorflow.org/api_docs/python/tf/math#Segmentation)\nfor an explanation of segments.\n\nThis operator is similar to the unsorted segment sum operator found\n[here](../../../api_docs/python/math_ops#UnsortedSegmentSum).\nInstead of computing the sum over segments, it computes the mean of all\nentries belonging to a segment such that:\n\n\\\\(output_i = 1/N_i \\\\sum_{j...} data\\[j...\\]\\\\) where the sum is over tuples\n`j...` such that `segment_ids[j...] == i` with \\\\N_i\\\\ being the number of\noccurrences of id \\\\i\\\\.\n\nIf there is no entry for a given segment ID `i`, it outputs 0.\n\nIf the given segment ID `i` is negative, the value is dropped and will not\nbe added to the sum of the segment.\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Args ---- ||\n|----------------|-----------------------------------------------------------------|\n| `data` | A `Tensor` with floating point or complex dtype. |\n| `segment_ids` | An integer tensor whose shape is a prefix of `data.shape`. |\n| `num_segments` | An integer scalar `Tensor`. The number of distinct segment IDs. |\n| `name` | A name for the operation (optional). |\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Returns ------- ||\n|---|---|\n| A `Tensor`. 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`. ||\n\n\u003cbr /\u003e"]]
TensorFlow 1 version
View source on GitHub