tf.sparse.segment_sqrt_n

Вычисляет сумму по разреженным сегментам тензора, делённую на sqrt(N).

tf.sparse.segment_sqrt_n(
    data,
    indices,
    segment_ids,
    num_segments=None,
    name=None,
    sparse_gradient=False
)

Прочитайте раздел о сегментации для объяснения сегментов.

Как и в tf.sparse.segment_mean, но вместо деления на размер сегмента, N, делим на sqrt(N) вместо этого.

Аргументы
`data`	A `Tensor` с данными, которые будут собраны в выводе.
`indices`	A 1-D `Tensor` с индексами в `data`. Имеет тот же ранг, что и `segment_ids`.
`segment_ids`	A 1-D `Tensor` с индексами в выходной `Tensor`. Значения должны быть отсортированы и могут повторяться.
`num_segments`	Необязательный int32 скаляр. Указывает размер выходного `Tensor`.
`name`	Название операции (необязательно).
`sparse_gradient`	Необязательный `bool`. По умолчанию `False`. Если `True`, градиент этой функции будет разреженным (`IndexedSlices`) вместо плотного (`Tensor`). Разреженный градиент будет содержать одну ненулевую строку для каждого уникального индекса в `indices`.

Возвращаемое значение
A `tensor` формы, как данные, за исключением измерения 0, которое имеет размер `k`, количество сегментов, указанных через `num_segments` или выведенных для последнего элемента в `segments_ids`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/sparse/segment_sqrt_n