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
tensorflow计算Loss
最新推荐文章于 2024-08-26 19:54:01 发布
本文介绍了如何在TensorFlow中计算损失(loss)。通过示例代码详细展示了权重的定义和损失函数的实现过程。
摘要由CSDN通过智能技术生成