机器学习

Tensorflow 打包-pack

  1. 用于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时

image-20191028151342313.png
当 axis= 1时
image-20191028153434573.png