1. 程式人生 > >tf.shape()與tensor.get_shape()

tf.shape()與tensor.get_shape()

兩種方法都可以獲得變數的shape

tf.shape(x)   其中x可以是tensor, 也可不是tensor,返回是一個tensor.

shape=tf.placeholder(tf.float32, shape=[None, 227, 227, 3]]

 我們經常會這樣來feed 資料,如果在執行的時候想知道None到底是多少,這時候只能通過tf.shape(x)[0]這種方式來獲得

tensor.get_shape()

只有tensor 有這個方法, 返回的是一個tuple