tf.raw_ops.SparseSegmentSumGradV2

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

Псевдонимы

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

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

tf.compat.v1.raw_ops.SparseSegmentSumGradV2

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

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

Аргументы
`grad`	A `Tensor`. Должен быть одним из следующих типов: `bfloat16`, `half`, `float32`, `float64`. Градиент, распространяемый к операции SparseSegmentSum.
`indices`	A `Tensor`. Должен быть одним из следующих типов: `int32`, `int64`. Индексы, переданные соответствующей операции SparseSegmentSum.
`segment_ids`	A `Tensor`. Должен быть одним из следующих типов: `int32`, `int64`. segment_ids, переданные соответствующей операции SparseSegmentSum.
`dense_output_dim0`	A `Tensor` типа `int32`. Измерение 0 от "data", переданное в операцию SparseSegmentSum.
`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/SparseSegmentSumGradV2