tf.compat.v1.sparse_segment_sqrt_n | TensorFlow v2.16.1 (original) (raw)

tf.compat.v1.sparse_segment_sqrt_n

Stay organized with collections Save and categorize content based on your preferences.

Computes the sum along sparse segments of a tensor divided by the sqrt(N).

Compat aliases for migration

tf.compat.v1.sparse.segment_sqrt_n

tf.compat.v1.sparse_segment_sqrt_n(
    data,
    indices,
    segment_ids,
    name=None,
    num_segments=None,
    sparse_gradient=False
)

N is the size of the segment being reduced.

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 assegment_ids.
segment_ids	A 1-D Tensor with indices into the output Tensor. Values should be sorted and can be repeated.
name	A name for the operation (optional).
num_segments	An optional int32 scalar. Indicates the size of the outputTensor.
sparse_gradient	An optional bool. Defaults to False. If True, the gradient of this function will be sparse (IndexedSlices) instead of dense (Tensor).

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 2024-04-26 UTC.