tf.math.cumprod
| Просмотреть исходный код на GitHub |
Вычислить кумулятивное произведение тензора x вдоль axis.
tf.math.cumprod(
x, axis=0, exclusive=False, reverse=False, name=None
)
По умолчанию этот оператор выполняет инклюзивное кумулятивное произведение, что означает, что первый элемент входного тензора идентичен первому элементу выходного тензора:
tf.math.cumprod([a, b, c]) # [a, a * b, a * b * c]
Установив параметр exclusive в значение True, вместо этого выполняется эксклюзивное кумулятивное произведение:
tf.math.cumprod([a, b, c], exclusive=True) # [1, a, a * b]
Установив параметр reverse в значение True, кумулятивное произведение выполняется в обратном направлении:
tf.math.cumprod([a, b, c], reverse=True) # [a * b * c, b * c, c]
Это более эффективно, чем использование отдельных операторов tf.reverse. Параметры reverse и exclusive также можно комбинировать:
tf.math.cumprod([a, b, c], exclusive=True, reverse=True) # [b * c, c, 1]
| Аргументы | |
|---|---|
x | Тензор. Допустимые типы: float32, float64, int64, int32, uint8, uint16, int16, int8, complex64, complex128, qint8, quint8, qint32, half. |
axis | Целое число (по умолчанию: 0). Должно быть в диапазоне [-rank(x), rank(x)). |
exclusive | Если True, выполнить эксклюзивное кумулятивное произведение. |
reverse | Булево (по умолчанию: False). |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Тензор. Имеет тот же тип, что и x. |
© 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/versions/r2.9/api_docs/python/tf/math/cumprod