tensorflow中读取张量的shape并且数据为int类型

遇到的问题：

batch_size = tf.shape(hidden1)[0]

如上所示，原有工程代码中是这样得到batch_size的，而这个batch_size是Tensor类型的，无法在np.zeros()或者for循环中使用。我查了很久如何将tensor转为int，并未找到相关方法。只找到了一大堆如何更改tensor中数据类型的博客，且大都是Ctrl+C/V，毫无营养。

解决办法：

后来发现，直接使用get_shape()方法可以直接得到shape的元组，如下所示：

batch_size = hidden_A.get_shape()[0]

直接在最开始得到batch_size的大小即可，然后就可以用在之后的for循环等需要值为Integer的地方了。

也许这个问题很基础，但我真的卡了好一段时间，在此简单记录，希望能帮到后来者。

Tensorflow中tf.shape(a)和a.get_shape可以参考这篇博客：链接，或者查阅官方文档。