Spec-Zone.ru › TensorFlow

tf.raw_ops.SparseAdd

Складывает два объекта SparseTensor, чтобы получить другой объект SparseTensor.

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

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

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

tf.compat.v1.raw_ops.SparseAdd

tf.raw_ops.SparseAdd(
    a_indices,
    a_values,
    a_shape,
    b_indices,
    b_values,
    b_shape,
    thresh,
    name=None
)

Предполагается, что индексы входных объектов SparseTensor упорядочены в стандартном лексикографическом порядке. Если это не так, перед этим шагом выполните SparseReorder, чтобы восстановить порядок индексов.

По умолчанию, если значения в некотором индексе в сумме дают ноль, выходной объект SparseTensor всё равно будет содержать это место в индексе, сохраняя ноль в соответствующей позиции значений. Чтобы изменить это поведение, вызывающий код может указать thresh: если модуль суммы будет строго меньше thresh, соответствующие значение и индекс не будут включены. В частности, thresh == 0 (значение по умолчанию) означает, что сохраняются все элементы, а фактическая фильтрация по порогу выполняется только для положительного значения.

В приведённых ниже формах nnz — это количество с учётом thresh.

Аргументы
a_indices Объект Tensor типа int64. Двумерный. indices первого SparseTensor, матрица размера [nnz, ndims].
a_values Объект Tensor. Должен иметь один из следующих типов: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, qint16, quint16, uint16, complex128, half, uint32, uint64. Одномерный. values первого SparseTensor, вектор размера [nnz].
a_shape Объект Tensor типа int64. Одномерный. shape первого SparseTensor, вектор размера [ndims].
b_indices Объект Tensor типа int64. Двумерный. indices второго SparseTensor, матрица размера [nnz, ndims].
b_values Объект Tensor. Должен иметь тот же тип, что и a_values. Одномерный. values второго SparseTensor, вектор размера [nnz].
b_shape Объект Tensor типа int64. Одномерный. shape второго SparseTensor, вектор размера [ndims].
thresh Объект Tensor. Должен иметь один из следующих типов: float32, float64, int32, uint8, int16, int8, int64, bfloat16, uint16, half, uint32, uint64. Скаляр. Порог модуля, определяющий, будет ли пара выходных значения и индекса занимать место.
name Имя операции (необязательно).
Возвращаемые значения
Кортеж объектов Tensor (sum_indices, sum_values, sum_shape).
sum_indices Объект Tensor типа int64.
sum_values Объект Tensor. Имеет тот же тип, что и a_values.
sum_shape Объект Tensor типа int64.

© 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/SparseAdd

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API