deep_learning_Function_tensorflow_unpack()

北战南征 提交于 2019-11-30 22:51:02

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

例如:

from tensorflow.models.rnn.ptb import reader
import 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)]
————————————————
原文链接:https://blog.csdn.net/UESTC_C2_403/article/details/72808556

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!