tf.unpack函数的用法

来源:互联网 发布:linux怎么打开防火墙 编辑:程序博客网 时间:2024/05/15 02:19

tf.unpack(A, axis)是一个解包函数。A是一个需要被解包的对象,axis是一个解包方式的定义,默认是零,如果是零,返回的结果就是按行解包。如果是1,就是按列解包。

例如:

from tensorflow.models.rnn.ptb import readerimport tensorflow as tf;import numpy as np;A = [[1, 2, 3], [4, 2, 3]]B = tf.unpack(A, axis=1)with tf.Session() as sess:print sess.run(B)

输出:

[array([1, 4], dtype=int32), array([2, 2], dtype=int32), array([3, 3], dtype=int32)]