tf.nn.softmax_cross_entropy_with_logits
此处的label的维度表示(1,样本数),label在tf.nn.softmax_cross_entropy_with_logits函数中会自动转化为独热矩阵,和logits做loss的运行。
tf.nn.softmax_cross_entropy_with_logits_v2
此处的label的维度表示(样本数,类别数目),就是tf.nn.softmax_cross_entropy_with_logits的label的独热矩阵形式。计算结果和前面一致。
tf.nn.softmax_cross_entropy_with_logits和tf.nn.softmax_cross_entropy_with_logits_v2用法总结
最新推荐文章于 2020-09-23 21:18:43 发布