代码之家  ›  专栏  ›  技术社区  ›  Gi Yeon Shin

从Tensorflow的多个热向量中提取索引

  •  0
  • Gi Yeon Shin  · 技术社区  · 6 年前

    例如,

    a = (1, 1, 0, 1, 0, 0, 0)
    

    那么,输出应该是

    output = (0, 1, 3)
    
    1 回复  |  直到 6 年前
        1
  •  0
  •   Mohan Radhakrishnan    6 年前

    如果你是这个意思

    a = tf.Variable(initial_value = [1, 1, 0, 1, 0, 0, 0])
    print( sess.run( tf.where(tf.equal(a, tf.constant(1)) )))
    

    [[0] [1]