Tensorflow 打包-pack

2020年5月4日

用于din_64.py

Reference

https://zhuanlan.zhihu.com/p/37637446
https://www.tensorflow.org/api_docs/python/tf/stack

Overview

理解pack与concat的区别：

tf.concat是沿某一维度拼接shape相同的张量，拼接生成的新张量维度不会增加。
而tf.stack是在新的维度上拼接，拼接后维度加1

例如：

import tensorflow as tf
a = tf.constant([[1,2,3],[4,5,6]]) 
b = tf.constant([[7,8,9],[10,11,12]])
ab1 = tf.concat([a,b],axis=0)
ab2 = tf.stack([a,b], axis=0)
sess = tf.Session()
print(sess.run(ab1)) 
print(sess.run(ab2))
print ab1
print ab2

当 axis= 0时

当 axis= 1时