DML_ELEMENT_WISE_ROUND_OPERATOR_DESC 구조체(directml.h)

아티클
11/12/2024

InputTensor 각 요소를 정수 값으로 반올림하여 결과를 OutputTensor해당 요소에 배치합니다.

이 연산자는 현재 위치 실행을 지원합니다. 즉, 바인딩하는 동안 OutputTensorInputTensor 별칭으로 허용됩니다.

통사론

struct DML_ELEMENT_WISE_ROUND_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  DML_ROUNDING_MODE     RoundingMode;
};

회원

InputTensor

형식: const DML_TENSOR_DESC*

읽을 입력 텐서입니다.

OutputTensor

형식: const DML_TENSOR_DESC*

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

RoundingMode

형식: DML_ROUNDING_MODE

둥글게 하는 방향을 결정하는 DML_ROUNDING_MODE.

DML_ROUNDING_MODE_HALVES_TO_NEAREST_EVEN경우: 값은 가장 가까운 정수로 반올림되고 중간 값(예: 0.5)은 가장 가까운 정수로 반올림됩니다.
DML_ROUNDING_MODE_TOWARD_ZERO경우: 값이 0으로 반올림됩니다. 이렇게 하면 소수 부분이 효과적으로 잘립니다.
DML_ROUNDING_MODE_TOWARD_INFINITY경우: 값이 가장 가까운 정수로 반올림되고 중간 값(예: 0.5)이 0에서 반올림됩니다(값의 부호에 따라 양수 또는 음의 무한대로).

가용도

이 연산자는 DML_FEATURE_LEVEL_2_1도입되었습니다.

Tensor 제약 조건

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

Tensor 지원

DML_FEATURE_LEVEL_3_0 이상

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

DML_FEATURE_LEVEL_2_1 이상

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

요구 사항

요구	값
지원되는 최소 클라이언트	Windows 10 빌드 20348
지원되는 최소 서버	Windows 10 빌드 20348
헤더	directml.h

다음을 통해 공유