tensorflow 如何看shape

分类: IT文章 • 2024-08-27 11:06:43

https://blog.****.net/yinxingtianxia/article/details/78121941

输入：

x= tf.truncated_normal([32, 32, 3], dtype=tf.float32)
 
print(tf.shape(x))
print(x.get_shape())
print(x.get_shape().as_list())

输出：

Tensor("Shape:0", shape=(3,), dtype=int32)
(32, 32, 3)
[32, 32, 3]