TensorFlow API 笔记

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/xinkuang126/article/details/88693196

Basic

tf.while_loop

def while_loop(cond, body, loop_vars):
	...
	while cond(loop_vars):
    	loop_vars = body(loop_vars)
    ...

loop_vars 参数先传入 cond 判断条件是否成立，如果 cond(loop_vars) 位 True，则把 loop_vars 参数传入 body 执行操作并用 body 操作后的返回值更新 loop_vars 参数，然后再次进入循环，直到不满足条件。

Graph

API

get tf.Operation from Graph

# session is tf.Session
# returns tf.Operation(in tensorflow/python/framework/ops.py)
session.graph.get_operations()

Model

analysis

tf.contrib.tfprof.model_analyzer.print_model_analysis()