Skip to main content

tf.unstack 用法

看代码:
import tensorflow as tf
import numpy as np
t=np.random.randint(1,10,(3,5))
ustack1=tf.unstack(t,axis=1)
ustack2=tf.unstack(t,axis=0)
sess=tf.Session()
print(t)
print(sess.run(ustack1))
print(sess.run(ustack2))
输出结果如下:
[[4 7 3 5 5] [3 5 8 4 1] [2 2 9 8 7]] [array([4, 3, 2]), array([7, 5, 2]), array([3, 8, 9]), array([5, 4, 8]), array([5, 1, 7])] [array([4, 7, 3, 5, 5]), array([3, 5, 8, 4, 1]), array([2, 2, 9, 8, 7])]

Comments

Popular posts from this blog

Session Run的用法

feed_dict参数的作用是替换图中的某个tensor的值。例如: a = tf.add(2, 5)                        #a=7 b = tf.multiply(a, 3)                 #b=3*7=21 with tf.Session() as sess:     print(sess.run(b))     replace_dict = {a:15}           #用15代替b算式中的a     print(sess.run(b, feed_dict = replace_dict)) --------------------- 输出如下: 21 45