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 ·
150 阅读 ·
0 评论