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 | Объект Tensor типа int64. 2-мерный. Индексы первого 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. 1-мерный. Значения первого SparseTensor, размер вектора [nnz]. |
a_shape | Объект Tensor типа int64. 1-мерный. Индексы первого SparseTensor, размер вектора [ndims]. |
b_indices | Объект Tensor типа int64. 2-мерный. Индексы второго SparseTensor, размер матрицы [nnz, ndims]. |
b_values | Объект Tensor. Должен иметь тот же тип, что и a_values. 1-мерный. Значения второго SparseTensor, размер вектора [nnz]. |
b_shape | Объект Tensor типа int64. 1-мерный. Индексы второго SparseTensor, размер вектора [ndims]. |
thresh | Объект Tensor. Должен быть одного из следующих типов: float32, float64, int32, uint8, int16, int8, int64, bfloat16, uint16, half, uint32, uint64. 0-мерное. Пороговое значение, определяющее, занимает ли пара значение/индекс выходного результата место. |
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