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

Возвращает
Кортеж объектов `Tensor` (output, sorted_unique_indices).
`output`	`Tensor`. Имеет тот же тип, что и `grad`.
`sorted_unique_indices`	`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