tf.raw_ops.SparseAdd
Добавляет два SparseTensor объекта, чтобы получить другой SparseTensor.
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 | A Tensor типа int64. 2-мерная. indices первого SparseTensor, размер [nnz, ndims] Матрица. |
a_values | A Tensor. Должен быть одним из следующих типов: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, uint16, complex128, half, uint32, uint64. 1-мерный. values первого SparseTensor, размер [nnz] Вектор. |
a_shape | A Tensor типа int64. 1-мерный. shape первого SparseTensor, размер [ndims] Вектор. |
b_indices | A Tensor типа int64. 2-мерная. indices второго SparseTensor, размер [nnz, ndims] Матрица. |
b_values | A Tensor. Должен иметь тот же тип, что и a_values. 1-мерный. values второго SparseTensor, размер [nnz] Вектор. |
b_shape | A Tensor типа int64. 1-мерный. shape второго SparseTensor, размер [ndims] Вектор. |
thresh | A Tensor. Должен быть одним из следующих типов: float32, float64, int32, uint8, int16, int8, int64, bfloat16, uint16, half, uint32, uint64. 0-мерный. Пороговое значение величины, определяющее, занимает ли пара значение/индекс выходного значения место. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Кортеж Tensor объектов (sum_indices, sum_values, sum_shape). | |
sum_indices | A Tensor типа int64. |
sum_values | A Tensor. Имеет тот же тип, что и a_values. |
sum_shape | A Tensor типа int64. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/SparseAdd