TensorFlow Eager笔记
入门教程中,定义loss和grad的计算函数如下:
def loss(model, x, y):
y_ = model(x)
return tf.losses.sparse_softmax_cross_entropy(labels=y, logits=y_)
def grad(model, inputs, targets):
with tf.GradientTape() as t...
原创
2018-11-12 23:11:37 ·
147 阅读 ·
0 评论