Compartilhar via


estrutura DML_MEAN_VARIANCE_NORMALIZATION2_OPERATOR_DESC (directml.h)

TBD

Importante

Essa API está disponível como parte do pacote redistribuível autônomo do DirectML (consulte Microsoft.AI.DirectML versão 1.15.0 e posterior. Consulte também de histórico de versões do DirectML.

Sintaxe

struct DML_MEAN_VARIANCE_NORMALIZATION2_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    _Maybenull_ const DML_TENSOR_DESC* ScaleTensor;
    _Maybenull_ const DML_TENSOR_DESC* BiasTensor;
    const DML_TENSOR_DESC* OutputTensor;
    UINT AxisCount;
    _Field_size_(AxisCount) const UINT* Axes;
    BOOL UseMean;
    BOOL UseVariance;
    FLOAT Epsilon;
    _Maybenull_ const DML_OPERATOR_DESC* FusedActivation;
};

Membros

InputTensor

Tipo: const DML_TENSOR_DESC*

Um tensor que contém os dados de entrada. As dimensões desse tensor devem ser { BatchCount, ChannelCount, Height, Width }.

ScaleTensor

Tipo: _Maybenull_ const DML_TENSOR_DESC*

Um tensor opcional que contém os dados de escala.

Se DML_FEATURE_LEVEL for menor que DML_FEATURE_LEVEL_4_0, as dimensões desse tensor deverão ser { ScaleBatchCount, ChannelCount, ScaleHeight, ScaleWidth }. As dimensões ScaleBatchCount, ScaleHeight e ScaleWidth devem corresponder a InputTensor ou ser definidas como 1 para transmitir automaticamente essas dimensões pela entrada.

Se DML_FEATURE_LEVEL for maior ou igual a DML_FEATURE_LEVEL_4_0, qualquer dimensão poderá ser definida como 1 e ser transmitida automaticamente para corresponder a InputTensor.

Se DML_FEATURE_LEVEL for menor que DML_FEATURE_LEVEL_5_2, esse tensor será necessário se BiasTensor estiver presente. Se DML_FEATURE_LEVEL for maior ou igual a DML_FEATURE_LEVEL_5_2, esse tensor poderá ser nulo, independentemente do valor de BiasTensor.

BiasTensor

Tipo: _Maybenull_ const DML_TENSOR_DESC*

Um tensor opcional que contém os dados de Viés.

Se DML_FEATURE_LEVEL for menor que DML_FEATURE_LEVEL_4_0, as dimensões desse tensor deverão ser { BiasBatchCount, ChannelCount, BiasHeight, BiasWidth }. As dimensões BiasBatchCount, BiasHeight e BiasWidth devem corresponder a InputTensor ou ser definidas como 1 para transmitir automaticamente essas dimensões pela entrada.

Se DML_FEATURE_LEVEL for maior ou igual a DML_FEATURE_LEVEL_4_0, qualquer dimensão poderá ser definida como 1 e ser transmitida automaticamente para corresponder a InputTensor.

Se DML_FEATURE_LEVEL for menor que DML_FEATURE_LEVEL_5_2, esse tensor será necessário se o ScaleTensor estiver presente. Se DML_FEATURE_LEVEL for maior ou igual a DML_FEATURE_LEVEL_5_2, esse tensor poderá ser nulo, independentemente do valor de ScaleTensor.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Um tensor para o qual gravar os resultados. As dimensões deste tensor são { BatchCount, ChannelCount, Height, Width }.

AxisCount

Tipo: UINT

O número de eixos. Esse campo determina o tamanho da matriz De eixos .

Axes

Tipo: _Field_size_(AxisCount) const UINT*

Os eixos ao longo dos quais calcular a Média e a Variação.

UseMean

Tipo: BOOL

TBD

UseVariance

Tipo: BOOL

TBD

Epsilon

Tipo: FLOAT

O valor de epsilon a ser usado para evitar divisão por zero. Um valor de 0,00001 é recomendado como padrão.

FusedActivation

Tipo: _Maybenull_ const DML_OPERATOR_DESC*

Uma camada de ativação fundida opcional a ser aplicada após a normalização.

Disponibilidade

Este operador foi introduzido no DML_FEATURE_LEVEL_6_3.

Restrições do Tensor

BiasTensor, InputTensor, OutputTensor e ScaleTensor devem ter o mesmo DataType e DimensionCount.

Suporte ao Tensor

Tensor Tipo Contagens de dimensões com suporte Tipos de dados com suporte
Tensor de entrada Entrada 1 a 8 FLOAT32, FLOAT16
Tensor de escala Entrada opcional 1 a 8 FLOAT32, FLOAT16
Tensor de viés Entrada opcional 1 a 8 FLOAT32, FLOAT16
Tensor de saída Saída 1 a 8 FLOAT32, FLOAT16