tensorflow中的函数获取Tensor维度的两种方法:
获取Tensor维度的两种方法:
Tensor.get_shape()
返回TensorShape对象,
如果需要确定的数值而把TensorShape当作list使用,肯定是不行的。
需要调用TensorShape的as_list()方法,
需要调用TensorShape.as_list()方法来获取维度数值。
来实践一下:
import tensorflow as tf
a = tf.zeros(shape=[10,20])
b = a.get_shape()
c = b.as_list()
print(b)
print(c)
输出结果:
(10, 20)
[10, 20]
Tensorflow.shape()
返回Tensor对象,需要调用Session.run()方法来获取维度数值。
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步