Tensorflow axis = -1 의미 파악하기


Tensorflow axis = -1 의미 파악하기

텐서플로우 페이지를 보다가 아래 코드를 발견하고 axis = -1 의 정체를 탐색해봤습니다 tf.nn.softmax(x,axis=-1) 정답부터 말하자면 우리가 파이썬의 리스트에서 이해하듯, axis = -1 은 주어진 배열의 마지막 axis 를 의미합니다 예시를 통해 보겠습니다 x라는 [2,3] shape의 텐서가 있을 때, 각각에 여러 max 관련 함수를 적용시켜보겠습니다. import tensorflow as tf x = tf.constant([[1., 2., 3.], [4., 5., 6.]]) 마지막 차원(default axis = -1)을 기준으로 softmax 함수를 적용한 모습입니다 tf.nn.softmax(x,axis=-1) 이는 tensorflow와 keras 에 적용한 결과값인데요 자세..


원문링크 : Tensorflow axis = -1 의미 파악하기