tensorflow计算Loss

小妖精Fsky

于 2017-02-24 10:06:13 发布

阅读量2.5k

点赞数 1

分类专栏： TensorFlow 文章标签： tensorflow

本文链接：https://blog.youkuaiyun.com/appleml/article/details/56832706

版权

本文介绍了如何在TensorFlow中计算损失(loss)。通过示例代码详细展示了权重的定义和损失函数的实现过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

    def loss(self, logits, labels):  #logit的shape= [batch]
        loss = None
        with tf.variable_scope('Hidden', reuse=True):
            W1 = tf.get_variable('W1')
            b1 = tf.get_variable('b1')
        with tf.variable_scope('Output', reuse=True):
            W2 = tf.get_variable('W2')
        loss2 = tf.nn.l2_loss(W1) + tf.nn.l2_loss(b1) + tf.nn

最低0.47元/天解锁文章