【TensorFlow】tf中关于shape的问题汇总

本文详细介绍了在 TensorFlow 中,tensor 的 static shape 和 dynamic shape 的概念及区别,包括 get_shape() 和 tf.shape() 的用法,以及 set_shape() 和 tf.reshape() 的功能差异,帮助理解形状操作。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

在TensorFlow中,涉及到shape的问题比较容易混淆,为此,在这里做一下区分,简单的汇总一下。

首先要说明一点,tf中tensor有两种shape,分别为static (inferred) shape 和 dynamic (true) shape,其中 static shape 用于构建图,是由创建这个tensor的op推断(inferred)得来,故又称 inferred shape。如果这个tensor的static shape未定义,则可用 tf.shape() 来获得其 dynamic shape。


1. 区分 x.get_shape() 和 x = tf.shape(x)

x.get_shape()返回 static shape,只有tensor有这个方法,返回的是元组。

x.get_shape().as_list() 是一个常用的方法,经常被用于将输出转为标准的python list。

关于 static shape 的样例如下所示:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值