Compartilhar via


estrutura DML_VALUE_SCALE_2D_OPERATOR_DESC (directml.h)

Executa uma função de dimensionamento e desvio elemento a elemento, Output = Scale * Input + Bias. Esse operador é semelhante ao uso de uma DML_ELEMENT_WISE_IDENTITY_OPERATOR_DESC com escala e desvio, exceto que DML_VALUE_SCALE_2D_OPERATOR_DESC aplica um desvio diferente para cada canal, em vez de um único desvio para todo o tensor.

Sintaxe

struct DML_VALUE_SCALE_2D_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Scale;
  UINT                  ChannelCount;
  const FLOAT           *Bias;
};

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 }.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Um tensor no qual gravar os resultados. As dimensões desse tensor devem corresponder às dimensões do InputTensor.

Scale

Tipo: FLOAT

Valor de escala a ser aplicado a todos os valores de entrada.

ChannelCount

Tipo: UINT

Esse campo determina o tamanho da matriz Bias. Esse campo deve ser definido como 1 ou 3 e também deve corresponder ao tamanho da dimensão Canal do tensor de entrada.

Bias

Tipo: const FLOAT*

Uma matriz de valores FLOAT que contém o termo de desvio para cada dimensão do tensor de entrada.

Disponibilidade

Esse operador foi introduzido no DML_FEATURE_LEVEL_1_0.

Restrições do Tensor

InputTensor e OutputTensor devem ter os mesmos DataType e Sizes.

Suporte ao Tensor

Tensor Tipo Contagens de dimensões com suporte Tipos de dados com suporte
InputTensor Entrada 4 FLOAT32, FLOAT16
OutputTensor Saída 4 FLOAT32, FLOAT16

Requisitos

Requisito Valor
Cabeçalho directml.h