tensorflow编程: Higher Order Functions

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]