1、tf.one_hot()

tf.one_hot(indices, depth, on_value, off_value, axis)

  • indices是一个列表,指定张量中独热向量的独热位置,或者说indeces是非负整数表示的标签列表。len(indices)就是分类的类别数。
  • tf.one_hot返回的张量的阶数为indeces的阶数+1。
  • 当indices的某个分量取-1时,即对应的向量没有独热值。
  • depth是每个独热向量的维度

  • on_value是独热值

  • off_value是非独热值

  • axis指定第几阶为depth维独热向量,默认为-1,即,指定张量的最后一维为独热向量

 

上一篇:webpack开发环境速度优化


下一篇:tensorflow机器学习中softmax_cross_entropy_with_logits和sigmoid_cross_entropy_with_logits的比较分析