Spec-Zone.ru › TensorFlow 2.4

tf.sparse.segment_sqrt_n

Просмотреть исходный код на GitHub

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

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

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

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

Аргументы
data A Tensor с данными, которые будут собраны в выходном значении.
indices Одномерный Tensor с индексами в data. Имеет тот же ранг, что и segment_ids.
segment_ids Одномерный Tensor с индексами в выходном Tensor. Значения должны быть отсортированы и могут повторяться.
num_segments Необязательный скаляр int32. Указывает размер выходного Tensor.
name Имя операции (необязательно).
Возвращает
A tensor формы как у данных, за исключением измерения 0, которое имеет размер k, число сегментов, указанных через num_segments или выведенных для последнего элемента в segments_ids.

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API