다음을 통해 공유


DML_ELEMENT_WISE_ROUND_OPERATOR_DESC 구조체(directml.h)

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에서 0으로 반올림됩니다(값의 부호에 따라 양수 또는 음의 무한대로).

가용성

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

텐서 제약 조건

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

텐서 지원

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