Spec-Zone.ru › TensorFlow 1.15

tf.compat.v2.reduce_sum

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

Просмотр псевдонимов

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

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

`tf.compat.v2.math.reduce_sum`

tf.compat.v2.reduce_sum(
    input_tensor, axis=None, keepdims=False, name=None
)

Суммирует input_tensor по измерениям, указанным в axis. Если keepdims не равно true, ранг тензора уменьшается на 1 для каждой записи в axis. Если keepdims равно true, уменьшенные измерения сохраняются с длиной 1.

Если axis равно None, все измерения сводятся, и возвращается тензор с одним элементом.

Например:

x = tf.constant([[1, 1, 1], [1, 1, 1]])
tf.reduce_sum(x)  # 6
tf.reduce_sum(x, 0)  # [2, 2, 2]
tf.reduce_sum(x, 1)  # [3, 3]
tf.reduce_sum(x, 1, keepdims=True)  # [[3], [3]]
tf.reduce_sum(x, [0, 1])  # 6
Аргументы
input_tensor Тензор для сворачивания. Должен иметь числовой тип.
axis Измерения для сворачивания. Если None (по умолчанию), сворачиваются все измерения. Должны быть в диапазоне [-rank(input_tensor), rank(input_tensor)).
keepdims Если true, сохраняет уменьшенные измерения с длиной 1.
name Имя операции (необязательно).
Возвращаемое значение
Свернутый тензор того же типа, что и входной тензор.

Совместимость с NumPy

Эквивалентно np.sum за исключением того, что NumPy преобразует uint8 и int32 в int64, в то время как TensorFlow возвращает тот же тип, что и входной.

© 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/r1.15/api_docs/python/tf/compat/v2/reduce_sum

Spec-Zone.ru

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