地址:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/ops/nn_ops.py 1688
函数定义:
def sparse_softmax_cross_entropy_with_logits(_sentinel=None, | |
labels=None,logits=None, | |
name=None): |
调用方法:
tf.nn.sparse_softmax_cross_entropy_with_logits(logits=py_x,labels=Y)
根据稀疏表示的label和输出层数据计算损失
参数1: labels,其值是稀疏表示的
labels向量大小[batch_size]
类型 `int32` or `int64`
取值范围 [0, num_classes)
参数2:logits,其表示隐藏层线性变换后非归一化后的结果
[batch_size, num_classes]
dtype `float32` or `float64`