tf.nn.sparse_softmax_cross_entropy_with_logits的用法

78 阅读 0 评论 52 点赞

我是靠谱客的博主稳重长颈鹿，这篇文章主要介绍tf.nn.sparse_softmax_cross_entropy_with_logits的用法，现在分享给大家，希望可以做个参考。

tf.nn.sparse_softmax_cross_entropy_with_logits(y, tf.argmax(y_)):

y表示的是实际类别，y_表示预测结果，这实际上面是把原来的神经网络输出层的softmax和cross_entrop何在一起计算，为了追求速度。

例如：

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import tensorflow as tf;    
import numpy as np;    
import matplotlib.pyplot as plt;    
    
y2 = tf.convert_to_tensor([[0, 0, 1, 0]], dtype=tf.int64)  
y_2 = tf.convert_to_tensor([[-2.6, -1.7, 3.2, 0.1]], dtype=tf.float32)  
c2 = tf.nn.sparse_softmax_cross_entropy_with_logits(y_2, tf.argmax(y2, 1))  
  
y3 = tf.convert_to_tensor([[0, 0, 1, 0], [0, 0, 1, 0]], dtype=tf.int64)  
y_3 = tf.convert_to_tensor([[-2.6, -1.7, -3.2, 0.1], [-2.6, -1.7, 3.2, 0.1]], dtype=tf.float32)  
c3 = tf.nn.sparse_softmax_cross_entropy_with_logits(y_3, tf.argmax(y3, 1))  
  
y4 = tf.convert_to_tensor([[0, 1, 0, 0]], dtype=tf.int64)  
y_4 = tf.convert_to_tensor([[-2.6, -1.7, -3.2, 0.1]], dtype=tf.float32)  
c4 = tf.nn.sparse_softmax_cross_entropy_with_logits(y_4, tf.argmax(y4, 1))  
  
with tf.Session() as sess:  
    print 'c2: ' , sess.run(c2)  
    print 'c3: ' , sess.run(c3)  
    print 'c4: ' , sess.run(c4)