tensorflow API：tf.unstack

最新推荐文章于 2022-01-25 13:51:20 发布

NockinOnHeavensDoor

最新推荐文章于 2022-01-25 13:51:20 发布

阅读量266

点赞数

分类专栏： tensorflow

本文链接：https://blog.csdn.net/NockinOnHeavensDoor/article/details/80714058

版权

tensorflow 专栏收录该内容

64 篇文章 0 订阅

订阅专栏

tf.unstack(
    value,
    num=None,
    axis=0,
    name='unstack'
)

作用：解包：

test = tf.random_normal([3,2],dtype=tf.float32)
with tf.Session() as sess:
    #默认axis=0
    print(sess.run([test,tf.unstack(test)]),"\n")
    print(sess.run([test,tf.unstack(test,axis=0)]),"\n")
    print(sess.run([test,tf.unstack(test,axis=1)]))

输出：
[array([[ 0.8021455 , 0.505679 ],
[ 1.8120127 , -0.03189123],
[ 1.1534904 , 0.9760273 ]], dtype=float32), [array([0.8021455, 0.505679 ], dtype=float32), array([ 1.8120127 , -0.03189123], dtype=float32), array([1.1534904, 0.9760273], dtype=float32)]]

[array([[-0.18855795, -0.8535253 ],
[ 0.7963146 , -0.245399 ],
[-0.53978336, 1.8063627 ]], dtype=float32), [array([-0.18855795, -0.8535253 ], dtype=float32), array([ 0.7963146, -0.245399 ], dtype=float32), array([-0.53978336, 1.8063627 ], dtype=float32)]]

[array([[ 1.3809487 , 0.4505079 ],
[ 0.5870948 , -0.7649997 ],
[-0.29083472, -0.8703685 ]], dtype=float32), [array([ 1.3809487 , 0.5870948 , -0.29083472], dtype=float32), array([ 0.4505079, -0.7649997, -0.8703685], dtype=float32)]]