Compartilhar via

estrutura DML_ELEMENT_WISE_ATAN_YX_OPERATOR_DESC (directml.h)

  • Artigo

Calcula o arcotangente de dois argumentos para cada elemento de ATensor e BTensor, em que ATensor é o eixo Y e BTensor é o eixo X, colocando o resultado no elemento correspondente de OutputTensor. Esse operador é indefinido para a origem (ou seja, quando ATensor e BTensor são 0 para elementos correspondentes).

GRU_Forward

f(y, x) = atan2(y, x)

Esse operador dá suporte à execução in-loco, o que significa que o tensor de saída tem permissão para alias ATensor ou BTensor durante a associação.

Sintaxe

struct DML_ELEMENT_WISE_ATAN_YX_OPERATOR_DESC {
  const DML_TENSOR_DESC *ATensor;
  const DML_TENSOR_DESC *BTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Membros

ATensor

Tipo: const DML_TENSOR_DESC*

O tensor de entrada do qual ler os valores do eixo Y.

BTensor

Tipo: const DML_TENSOR_DESC*

O tensor de entrada do qual ler os valores do eixo X.

OutputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de saída para o qual gravar os resultados.

Comentários

Disponibilidade

Esse operador foi introduzido em DML_FEATURE_LEVEL_3_1.

Restrições do Tensor

ATensor, BTensor e OutputTensor devem ter os mesmos DataType, DimensionCount e Sizes.

Suporte ao Tensor

Tensor Tipo Contagens de dimensões com suporte Tipos de dados com suporte
Atensor Entrada 1 a 8 FLOAT32, FLOAT16
BTensor Entrada 1 a 8 FLOAT32, FLOAT16
OutputTensor Saída 1 a 8 FLOAT32, FLOAT16

Requisitos

Requisito Valor
Cliente mínimo com suporte Windows Build 22000
Servidor mínimo com suporte Windows Build 22000
Cabeçalho directml.h