参考自https://blog.csdn.net/feynman233/article/details/79187304,记录下方便之后查看。
print np.sum([np.prod(v.get_shape().as_list()) for v in tf.trainable_variables()])
参考自https://blog.csdn.net/feynman233/article/details/79187304,记录下方便之后查看。
print np.sum([np.prod(v.get_shape().as_list()) for v in tf.trainable_variables()])
转载于:https://my.oschina.net/u/4017456/blog/2906986