python-tensorflow 控制依赖案例（阶乘）

#%%

import tensorflow as tf
import numpy as np
import os
import hashlib
import matplotlib.pylab as plt
import cv2
from PIL import Image

#%%
#1、定义一个变量
sum = tf.Variable(initial_value=1,dtype=tf.int32)

#2、定义一个占位符
i= tf.Variable(initial_value=6,dtype=tf.int32)

#3、更新操作
temp_sum = sum*i
assign_sub_op = tf.assign_sub(i,1)
assign_op     = tf.assign(sum,temp_sum)

#如果需要执行这个代码块中的内容，必须先执行control_dependencies中给定的操作/tensor
with tf.control_dependencies([assign_op,assign_sub_op]):
    temp_sum = tf.identity(temp_sum)

with tf.Session() as sess:
    tf.global_variables_initializer().run()
    
    a = sess.run(i)
    for j in range(1,a):
        #执行更新操作
        print('temp_sum:',sess.run(temp_sum))

输出：

 

temp_sum: 5

temp_sum: 20

temp_sum: 60

temp_sum: 120

temp_sum: 120