TensorFlow之tf.unstack学习循环神经网络中用到!

Posted 故笙

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了TensorFlow之tf.unstack学习循环神经网络中用到!相关的知识,希望对你有一定的参考价值。

unstack(
    value,
    num=None,
    axis=0,
    name=‘unstack‘
)

tf.unstack()  

  将给定的R维张量拆分成R-1维张量

  value根据axis分解成num个张量,返回的值是list类型,如果没有指定num则根据axis推断出!

DEMO:

import tensorflow as tf
a = tf.constant([3,2,4,5,6])
b = tf.constant([1,6,7,8,0])
c = tf.stack([a,b],axis=0)
d = tf.stack([a,b],axis=1)
e = tf.unstack([a,b],axis=0)
f = tf.unstack([a,b],axis=1)

with tf.Session() as sess:
    print(sess.run(c))
    print(sess.run(d))
    print(sess.run(e))
    print(sess.run(f))

  输出:

[[3 2 4 5 6]
[1 6 7 8 0]]

--------------------
[[3 1]
[2 6]
[4 7]
[5 8]
[6 0]]

----------------------
[array([3, 2, 4, 5, 6]), array([1, 6, 7, 8, 0])]

----------------------
[array([3, 1]), array([2, 6]), array([4, 7]), array([5, 8]), array([6, 0])]

 

以上是关于TensorFlow之tf.unstack学习循环神经网络中用到!的主要内容,如果未能解决你的问题,请参考以下文章

tf.unstack()tf.stack()

使用 one-hot 代码的 TensorFlow 混淆矩阵

使用data_flow_ops构造batch数据集

Tensor的合并与分割

TensorFlow框架之RNN循环神经网络详解

Tensorflow 2.0的自定义训练循环的学习率