tensorflow计算Loss

最新推荐文章于 2022-04-07 14:54:02 发布

小妖精Fsky

最新推荐文章于 2022-04-07 14:54:02 发布

阅读量2.4k

点赞数 1

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

本文链接：https://blog.csdn.net/appleml/article/details/56832706

版权

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

摘要由CSDN通过智能技术生成

    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