地址:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/ops/nn_ops.py 1589
函数定义:
def softmax_cross_entropy_with_logits(_sentinel=None, | |
labels=None, logits=None, | |
dim=-1, name=None): |
函数调用:
tf.nn.softmax_cross_entropy_with_logits(labels=Y, logits=py_x)
函数说明
参数1:labels
[batch_size, num_classes]
类型: `float16`, `float32`, or `float64`
参数2:logits
[batch_size, num_classes]
类型 `float16`, `float32`, or `float64`