VGGNet-16中的一些函數解析

使用get_shape()[-1].value獲取tensor的最後一維的長度,get_shape()[0].value獲取tensor第一維的長度。例如:python input = tf.random_normal([224,225,3], dtype=tf.float32,stddev = 1e-1) >>> input.get_shape()[0].value 224 >>> input.
相關文章
相關標籤/搜索