# tf.shape() 返回的是tensor的维度
代码如下:
import tensorflow as tf #导入tensorflow包
# 定义一个tensor,是由2行4列构成 类型为:float32
constant_tensor=tf.constant(value=[[1.1,2.2,3.3,4.4],[4.1,3.1,2.1,1.1]], dtype=tf.float32)
# shape函数是现实tensor的维度结构
shape_constant_tensor=tf.shape(constant_tensor)
# 定义一个Session
sess=tf.Session()
# 运行这个Session
sess.run(shape_constant_tensor)
输出结果:
array([2, 4]
恭喜您学习完成!
网友评论