tf.raw_ops.SparseReduceSum
Вычисляет сумму элементов по измерениям SparseTensor.
tf.raw_ops.SparseReduceSum(
input_indices, input_values, input_shape, reduction_axes, keep_dims=False,
name=None
)
Этот оператор принимает SparseTensor и является разреженным аналогом tf.reduce_sum(). В частности, этот оператор также возвращает плотное Tensor вместо разреженного.
Суммирует sp_input по измерениям, указанным в reduction_axes. Если keep_dims не равно true, ранг тензора уменьшается на 1 для каждого значения в reduction_axes. Если keep_dims равно true, уменьшенные измерения сохраняются с длиной 1.
Если у reduction_axes нет элементов, все измерения уменьшаются, и возвращается тензор с одним элементом. Кроме того, оси могут быть отрицательными, которые интерпретируются в соответствии с правилами индексирования в Python.
| Аргументы | |
|---|---|
input_indices | A Tensor типа int64. 2-мерная. N x R матрица с индексами ненулевых значений в SparseTensor, возможно, не в каноническом порядке. |
input_values | A Tensor. Должен быть одного из следующих типов: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, uint16, complex128, half, uint32, uint64. 1-мерный. N ненулевые значения, соответствующие input_indices. |
input_shape | A Tensor типа int64. 1-мерный. Формат входного SparseTensor. |
reduction_axes | A Tensor типа int32. 1-мерный. Вектор длины K, содержащий оси уменьшения. |
keep_dims | Необязательный bool. По умолчанию False. Если true, сохраняются уменьшенные измерения с длиной 1. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
A Tensor. Имеет тот же тип, что и input_values. |
© 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/SparseReduceSum