DML_ACTIVATION_HARDMAX_OPERATOR_DESC 구조체(directml.h)

InputTensor의 각 요소에 대해 hardmax 함수를 수행하여 결과를 OutputTensor의 해당 요소에 배치합니다.

연산자는 지정된 입력에서 각 행의 hardmax(계층에서 가장 큰 값이 처음 나타나는 경우 1, 다른 모든 값의 경우 0)를 계산합니다.

구문

struct DML_ACTIVATION_HARDMAX_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*

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

설명

연산자는 지정된 입력의 일괄 처리에서 각 계층에 대한 hardmax(첫 번째 최대값의 경우 1, 다른 모든 값의 경우 0)를 계산합니다. 입력은 크기가 2D 텐서(Tensor)(batch_size x input_feature_dimensions)입니다. 출력 텐서의 모양은 동일하며 해당 입력의 hardmax 값을 포함합니다.

가용성

이 연산자는 에서 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_ARGMAX_OPERATOR_DESC 구조체