DML_ACTIVATION_SHRINK_OPERATOR_DESC 구조체(directml.h)

아티클
03/05/2024

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

f(x) = x - Bias, if x > Threshold
       x + Bias, if x < -Threshold
       0,        otherwise

이 연산자는 현재 위치 실행을 지원합니다. 즉, 출력 텐서가 바인딩 중에 InputTensor 별칭을 지정하도록 허용됩니다.

구문

struct DML_ACTIVATION_SHRINK_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Bias;
  FLOAT                 Threshold;
};

멤버

InputTensor

형식: const DML_TENSOR_DESC*

읽을 입력 텐서입니다.

OutputTensor

형식: const DML_TENSOR_DESC*

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

Bias

형식: FLOAT

바이어스의 값입니다. 이 값의 일반적인 기본값은 0.0입니다.

Threshold

형식: FLOAT

임계값입니다. 이 값의 일반적인 기본값은 0.5입니다.

설명

가용성

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

텐서 제약 조건

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

텐서 지원

DML_FEATURE_LEVEL_5_1 이상

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

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

요구 사항

요구 사항	값
지원되는 최소 클라이언트	Windows 10 버전 2004(10.0; 빌드 19041)
지원되는 최소 서버	Windows Server 버전 2004(10.0; 빌드 19041)
머리글	directml.h

Share via