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`	`Tensor` с данными, которые будут собраны в выходном результате.
`indices`	Одномерный `Tensor` с индексами в `data`. Имеет тот же ранг, что и `segment_ids`.
`segment_ids`	Одномерный `Tensor` с индексами в выходном `Tensor`. Значения должны быть отсортированы; допускаются повторы.
`num_segments`	Необязательный скаляр int32. Указывает размер выходного `Tensor`.
`name`	Имя операции (необязательно).
`sparse_gradient`	Необязательный `bool`. По умолчанию — `False`. Если `True`, градиент этой функции будет разреженным (`IndexedSlices`), а не плотным (`Tensor`). Разреженный градиент будет содержать одну ненулевую строку для каждого уникального индекса в `indices`.

Возвращаемое значение
`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