python - TensorFlow中有没有相当于python的reduce函数？

Question

我想要 TensorFlow 中的函数与 Pythion reduce() 具有相同的效果

例如，如果我有一个值为 [a1, a2, a3] 的张量 a 和一个函数 func()，我想要[func(a1, a2), func(func(a1, a2), a3)]。如果 a 是一个 Python 列表，我可以简单地执行

from functools import reduce
# a = [a1, a2, a3]
A = reduce(func, a)

如果a是TF对象的张量(不是Python列表!)并且我想要相同的A怎么办？如果没有替代的 TF 功能，我该如何有效地做到这一点？

Answer 1

如果你实际上指的是减少，那么相当于 tensorflow 是 tf.foldl和 tf.foldr 。示例:

elems = tf.constant(["hello", "my", "name", "is", "inigo", "montoya"])
tf.foldl(lambda a, x: a +" " + x, elems)
#<tf.Tensor: shape=(), dtype=string, numpy=b'hello my name is inigo montoya'>
tf.foldr(lambda a, x: a +" " + x, elems)
#<tf.Tensor: shape=(), dtype=string, numpy=b'montoya inigo is name my hello'>

但是，根据您的示例，您可能需要 tf.scan :

tf.scan(lambda a, x: a +" " + x, elems)
<tf.Tensor: shape=(6,), dtype=string, numpy=
    ([b'hello', b'hello my', b'hello my name', b'hello my name is',
    b'hello my name is inigo', b'hello my name is inigo montoya'], dtype=object)>