Higher Order Operators
tf.scan
从 elems列表 中 依次 扫描读取 元素 放入 公式进行 迭代计算。相当于python的 map 函数。
tf.scan (fn, elems, initializer=None, parallel_iterations=10, back_prop=True, swap_memory=False, infer_shape=True, name=None)
import numpy as np
import tensorflow as tf
elems = np.array([1, 2, 3, 4, 5, 6])
sum_e = tf.scan(lambda a, b: a + b, elems)
sess = tf.InteractiveSession()
print sum_e.eval()
[ 1 3 6 10 15 21]