tf.sparse.segment_sqrt_n
Computes the sum along sparse segments of a tensor divided by the sqrt(N).
tf.sparse.segment_sqrt_n(
data, indices, segment_ids, num_segments=None, name=None
)
Read the section on
segmentation
for an explanation of segments.
Like tf.sparse.segment_mean
, but instead of dividing by the size of the
segment, N
, divide by sqrt(N)
instead.
Args |
data
|
A Tensor with data that will be assembled in the output.
|
indices
|
A 1-D Tensor with indices into data . Has same rank as
segment_ids .
|
segment_ids
|
A 1-D Tensor with indices into the output Tensor . Values
should be sorted and can be repeated.
|
num_segments
|
An optional int32 scalar. Indicates the size of the output
Tensor .
|
name
|
A name for the operation (optional).
|
Returns |
A tensor of the shape as data, except for dimension 0 which
has size k , the number of segments specified via num_segments or
inferred for the last element in segments_ids .
|
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 2023-03-17 UTC.
[null,null,["Last updated 2023-03-17 UTC."],[],[]]