卷积核weights参数shape说明

weights = tf.get_variable('weights',
                                  shape=[3, 3, 3, 16],
                                  dtype=tf.float32,
                                  initializer=tf.truncated_normal_initializer(stddev=0.1, dtype=tf.float32))

shape为4维tensor,各参数含义:[width, height, channels, kernel_nums]

猜你喜欢

转载自blog.csdn.net/oMoDao1/article/details/81974687