TensorFlow获取Tensor维度

最新推荐文章于 2024-04-16 09:44:38 发布

littlehaes

最新推荐文章于 2024-04-16 09:44:38 发布

阅读量1.6k

点赞数

分类专栏：深度学习文章标签： Tensor TensorFlow

本文链接：https://blog.csdn.net/littlehaes/article/details/81513572

版权

本文介绍了在TensorFlow中获取Tensor维度的三种方法：通过a.shape、a.get_shape()和tf.shape(a)，并解释了动态shape与静态shape的区别，特别是在reshape操作中的应用场景。

摘要由CSDN通过智能技术生成

Welcome To My Blog

获取Tensor维度

比如一个Tensor为a = tf.constant([[1,2],[3,4]],name='a'),有三种方式可以获取a的维度
1. a.shape
2. a.get_shape()
3. tf.shape(a)
前两种返回类型是TensorShape,代表静态shape,a.shape.as_list()返回list类型的shape
第三种返回类型是Tensor,代表动态shape

动态shape与静态shape

当需要reshape Tensor时,就能体现出差异了

import tensorflow as tf
b = tf.placeholder(tf.float32,[None,10,32])
b_static = b.shape.as_list()
b_dynamic = tf.unstack(tf.shape(b))
dim = [s[1] if s[0] if