tf.raw_ops.SparseSegmentMean
bookmark_border bookmark
Stay organized with collections
Save and categorize content based on your preferences.
Computes the mean along sparse segments of a tensor.
View aliases
Compat aliases for migration
See
Migration guide for
more details.
tf.compat.v1.raw_ops.SparseSegmentMean
tf . raw_ops . SparseSegmentMean (
data , indices , segment_ids , sparse_gradient = False , name = None
)
See tf.sparse.segment_sum
for usage examples.
Like SegmentMean
, but segment_ids
can have rank less than data
's first
dimension, selecting a subset of dimension 0, specified by indices
.
Args
data
A Tensor
. Must be one of the following types: bfloat16
, half
, float32
, float64
.
indices
A Tensor
. Must be one of the following types: int32
, int64
.
A 1-D tensor. Has same rank as segment_ids
.
segment_ids
A Tensor
. Must be one of the following types: int32
, int64
.
A 1-D tensor. Values should be sorted and can be repeated.
sparse_gradient
An optional bool
. Defaults to False
.
name
A name for the operation (optional).
Returns
A Tensor
. Has the same type as data
.
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 2024-01-23 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 2024-01-23 UTC."],[],[]]