tf.raw_ops.SparseSegmentMeanGradV2

Вычисляет градиенты для SparseSegmentMean.

Псевдонимы

Псевдонимы для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.raw_ops.SparseSegmentMeanGradV2

tf.raw_ops.SparseSegmentMeanGradV2(
    grad, indices, segment_ids, dense_output_dim0, name=None
)

Возвращает тензор "output" с такой же формой, как grad, за исключением измерения 0, значение которого равно числу уникальных индексов в "indices". Также возвращает вектор "sorted_unique_indices", содержащий соответствующие индексы из "indices".

Аргументы
`grad`	A `Tensor`. Должен быть одним из следующих типов: `bfloat16`, `half`, `float32`, `float64`. градиент, проходящий в операцию SparseSegmentMean.
`indices`	A `Tensor`. Должен быть одним из следующих типов: `int32`, `int64`. индексы, переданные соответствующей операции SparseSegmentMean.
`segment_ids`	A `Tensor`. Должен быть одним из следующих типов: `int32`, `int64`. segment_ids, переданные соответствующей операции SparseSegmentMean.
`dense_output_dim0`	A `Tensor` типа `int32`. измерение 0 "data", переданное в операцию SparseSegmentMean.
`name`	Имя операции (необязательно).

Возвращаемое значение
Кортеж из объектов `Tensor` (output, sorted_unique_indices).
`output`	A `Tensor`. Имеет тот же тип, что и `grad`.
`sorted_unique_indices`	A `Tensor`. Имеет тот же тип, что и `indices`.

© 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/raw_ops/SparseSegmentMeanGradV2