tensorflow 如何看shape

https://blog.csdn.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]

猜你喜欢

转载自www.cnblogs.com/ymjyqsx/p/10863320.html