DML_ACTIVATION_SOFTMAX_OPERATOR_DESC 구조체(directml.h)

InputTensor에서 softmax 활성화 함수를 수행하여 결과를 OutputTensor의 해당 요소에 배치합니다.

For 1-D InputTensor:
// Let x[i] be the current element in the InputTensor, and j be the total number of elements in the InputTensor
f(x[i]) = exp(x[i]) / sum(exp(x[0]), ..., exp(x[j-1]))

여기서 exp(x)는 자연 지수 함수입니다.

구문

struct DML_ACTIVATION_SOFTMAX_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

멤버

InputTensor

형식: const DML_TENSOR_DESC*

읽을 입력 텐서입니다. 이 텐서는 유효 순위 가 2보다 크지 않아야 합니다. 텐서의 유효 순위는 크기 1의 맨 왼쪽 차원을 제외하고 텐서의 DimensionCount 입니다. 예를 들어 의 { 1, 1, BatchCount, Width } 텐서 크기는 유효하며 크기가 인 텐서와 같습니다.{ BatchCount, Width }

OutputTensor

형식: const DML_TENSOR_DESC*

결과를 쓸 출력 텐서입니다.

가용성

이 연산자는 에서 DML_FEATURE_LEVEL_1_0도입되었습니다.

텐서 제약 조건

InputTensor 및 OutputTensor 에는 동일한 DataType, DimensionCount 및 크기가 있어야 합니다.

텐서 지원

DML_FEATURE_LEVEL_3_0 이상

텐서	종류	지원되는 차원 수	지원되는 데이터 형식
InputTensor	입력	1~8개	FLOAT32, FLOAT16
OutputTensor	출력	1~8개	FLOAT32, FLOAT16

DML_FEATURE_LEVEL_2_0 이상

텐서	종류	지원되는 차원 수	지원되는 데이터 형식
InputTensor	입력	4~5개	FLOAT32, FLOAT16
OutputTensor	출력	4~5개	FLOAT32, FLOAT16

DML_FEATURE_LEVEL_1_0 이상

텐서	종류	지원되는 차원 수	지원되는 데이터 형식
InputTensor	입력	4	FLOAT32, FLOAT16
OutputTensor	출력	4	FLOAT32, FLOAT16

요구 사항

요구 사항	값
헤더	directml.h

추가 정보

DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC 구조체

DML_ARGMAX_OPERATOR_DESC 구조체