最近在學習TensorFlow的時候遇到了許多問題，現一一記錄，幫助自己記憶的同時也希望能幫到大家。

正文-------------------------------------------------

做分類任務的時候，經常會使用交叉熵函式作為loss來計算：形式如下

def evaluation(logits, labels):
    with tf.variable_scope('accuracy') as scope:
        correct_prediction = tf.equal(tf.argmax(labels, 1), tf.argmax(logits, 1))
        correct = tf.cast(correct_prediction, tf.float32)
        accuracy = tf.reduce_mean(correct)

    return accuracy
 

但是，如果這樣寫的話，他會報錯，如標題那樣，說是axis希望在[-1, 1)之間，而不是1，，，，，，，，，，，

這個東西我查了一下，意思是說，axis必須是-1到1之間，並且左閉右開，取值為1是不行的，

於是 將上面程式碼中的 1改成-1，就可以了！

如下。。。。。。。。。

correct_prediction = tf.equal(tf.argmax(labels, -1), tf.argmax(logits, -1))