tf.math.accumulate_n
Возвращает поэлементную сумму списка тензоров. (устаревшее)
tf.math.accumulate_n(
inputs, shape=None, tensor_dtype=None, name=None
)
Необязательно передать shape и tensor_dtype для проверки формы и типа, в противном случае они выводятся.
Например:
a = tf.constant([[1, 2], [3, 4]])
b = tf.constant([[5, 0], [0, 6]])
tf.math.accumulate_n([a, b, a]).numpy()
array([[ 7, 4],
[ 6, 14]], dtype=int32)# Explicitly pass shape and type
tf.math.accumulate_n(
[a, b, a], shape=[2, 2], tensor_dtype=tf.int32).numpy()
array([[ 7, 4],
[ 6, 14]], dtype=int32)
Примечание: Вход должен быть списком или кортежем. Данная функция не обрабатывает IndexedSlices
См. также:
-
tf.reduce_sum(inputs, axis=0)- Это выполняет ту же математическую операцию, ноtf.add_nможет быть более эффективным, потому что суммирует тензоры напрямую.reduce_sum, с другой стороны, вызываетtf.convert_to_tensorдля списка тензоров, ненужно складывая их в один тензор перед суммированием. -
tf.add_n- Это другой оберточный Python для той же операции. Он имеет практически идентичную функциональность.
| Аргументы | |
|---|---|
inputs | Список объектов Tensor, каждый с одинаковой формой и типом. |
shape | Ожидаемая форма элементов inputs (необязательно). Также контролирует форму выходных данных этой операции, что может повлиять на вывод типа в других операциях. Значение None означает «вывести форму входных данных из форм в inputs». |
tensor_dtype | Ожидаемый тип данных inputs (необязательно). Значение None означает «вывести тип данных ввода из inputs[0]». |
name | Название операции (необязательно). |
| Возвращает | |
|---|---|
Тензор той же формы и типа, что и элементы inputs. |
| Возбуждает | |
|---|---|
ValueError | Если inputs не имеют одинаковой формы и типа или форма не может быть выведена. |
© 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/math/accumulate_n