tf.math.unsorted_segment_mean
| Просмотреть исходный код на GitHub |
Вычисляет среднее значение по сегментам тензора.
tf.math.unsorted_segment_mean(
data, segment_ids, num_segments, name=None
)
Прочитайте раздел о сегментации для объяснения сегментов.
Этот оператор аналогичен оператору суммирования по несортированным сегментам, который можно найти здесь. Вместо вычисления суммы по сегментам, он вычисляет среднее значение всех элементов, принадлежащих сегменту, таким образом:
\(output_i = 1/N_i \sum_{j...} data[j...]\) где сумма выполняется по наборам j... таким образом, что segment_ids[j...] == i, где \N_i\ - это количество вхождений id \i\.
Если для данного идентификатора сегмента i, отсутствуют элементы, то он возвращает 0.
Если указанный идентификатор сегмента i отрицательный, значение отбрасывается и не добавляется в сумму сегмента.
| Аргументы | |
|---|---|
data | A Tensor с плавающей точкой или комплексным типом данных. |
segment_ids | Целочисленный тензор, форма которого является префиксом data.shape. |
num_segments | Целочисленная скалярная величина Tensor. Количество различных идентификаторов сегментов. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
A Tensor. Имеет ту же форму, что и данные, за исключением первых segment_ids.rank измерений, которые заменяются одним измерением с размером num_segments. |
© 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/r1.15/api_docs/python/tf/math/unsorted_segment_mean
