DML_LP_NORMALIZATION_OPERATOR_DESC structure (directml.h)

Performs an Lp-normalization function along the specified axis of the input tensor.

Syntax

struct DML_LP_NORMALIZATION_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  UINT                  Axis;
  FLOAT                 Epsilon;
  UINT                  P;
};

Members

InputTensor

Type: const DML_TENSOR_DESC*

The tensor containing the input data.

OutputTensor

Type: const DML_TENSOR_DESC*

The tensor to write the results to. This tensor's Sizes should match the InputTensor.

Axis

Type: UINT

The axis on which to apply normalization.

Epsilon

Type: FLOAT

The epsilon value to use to avoid division by zero. A value of 0.00001 is recommended as default.

P

Type: UINT

The order of the normalization (either 1 or 2).

Availability

This operator was introduced in DML_FEATURE_LEVEL_1_0.

Tensor constraints

InputTensor and OutputTensor must have the same DataType, DimensionCount, and Sizes.

Tensor support

DML_FEATURE_LEVEL_3_1 and above

Tensor Kind Supported dimension counts Supported data types
InputTensor Input 1 to 8 FLOAT32, FLOAT16
OutputTensor Output 1 to 8 FLOAT32, FLOAT16

DML_FEATURE_LEVEL_1_0 and above

Tensor Kind Supported dimension counts Supported data types
InputTensor Input 4 FLOAT32, FLOAT16
OutputTensor Output 4 FLOAT32, FLOAT16

Requirements

Requirement Value
Header directml.h