tf.raw_ops.SparseSegmentSqrtNGradV2

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

Просмотреть псевдонимы

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

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.SparseSegmentSqrtNGradV2

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

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

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