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