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.
相关文章
相关标签/搜索