tf.nn

来源:互联网 发布:selected知乎 编辑:程序博客网 时间:2024/04/27 22:50

tf.nn.sparse_softmax_cross_entropy_with_logits(logits, labels)

  1. 针对硬分类;
  2. logits是unscale的,其shape是[batch_size, num_classes],dtype为float32 or float64;
  3. labels的shape是[batch_size],dtype为int64,值的含义是所属的类。
原创粉丝点击