tensorflow
猿球崛起
这个作者很懒,什么都没留下…
展开
-
tf.nn.moments笔记: axes到底咋回事?
x = np.arange(12,dtype=np.float32).reshape(3,4)a = tf.nn.moments(tf.constant(x),[0])b = tf.nn.moments(tf.constant(x),[1])# 相当于展开[0,1]维度, 在所有数上计算均值和方差c = tf.nn.moments(tf.constant(x),[0,1])d = tf....原创 2019-09-19 11:34:43 · 763 阅读 · 1 评论 -
tf.gradients/optimizer.compute_gradients/optimizer.apply_gradients 联合使用极简demo
代码tf.reset_default_graph()optim = tf.train.AdamOptimizer(0.01)a = tf.Variable(initial_value=0.,name='a')b = 2*ay = a+btvars = tf.trainable_variables()# grads = tf.gradients(y,tvars)# app = op...原创 2019-09-19 14:26:43 · 3311 阅读 · 0 评论