tf.foldl

import tensorflow as tf
elems = tf.constant([1, 2, 3, 4, 5, 6])
sum1 = tf.foldl(lambda a, x: a * x, elems)
print(sum1.numpy())


sum2 = tf.foldl(lambda a, x: a + x, elems)
print(sum2.numpy())
720
21
posted @ 2022-08-19 22:51  luoganttcc  阅读(32)  评论(0)    收藏  举报