Spec-Zone.ru › TensorFlow

tf.raw_ops.SparseReduceSum

Вычисляет сумму элементов по измерениям SparseTensor.

Псевдонимы

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

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.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 Tensor типа int64. 2-мерная. 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-мерный. N ненулевых значений, соответствующих input_indices.
input_shape Tensor типа int64. 1-мерный. Формат входного SparseTensor.
reduction_axes Tensor типа int32. 1-мерный. Вектор длины 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

Spec-Zone.ru

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