我正在编写自己的功能层,比如
def vgg_block(x, is_training, output_depth,... name = "vgg_block"):
with tf.variable_scope(name):
if pool_size is not None:
x = tf.layers.max_pooling2d(x, pool_size=pool_size, strides=pool_strides, padding=pool_padding)
for i_conv in range(n_c