estrutura DML_TILE_OPERATOR_DESC (directml.h)

Artigo
08/23/2023

Constrói um tensor de saída ao lado do tensor de entrada. Os elementos em cada dimensão do tensor de entrada são repetidos por um múltiplo na matriz Repeats .

Sintaxe

struct DML_TILE_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  UINT                  RepeatsCount;
  const UINT            *Repeats;
};

Membros

InputTensor

Tipo: const DML_TENSOR_DESC*

O tensor do qual ler, que contém os elementos a serem lado a lado.

OutputTensor

Tipo: const DML_TENSOR_DESC*

O tensor no qual gravar, que conterá a saída lado a lado. Para cada dimensão i em [0, InputTensor.DimensionCount-1], o tamanho da saída é calculado como OutputTensor.Sizes[i] = InputTensor.Sizes[i] * Repeats[i]. Esse tensor deve ter o mesmo DimensionCount que o tensor de entrada.

RepeatsCount

Tipo: UINT

Esse campo determina o tamanho da matriz Repeats . Esse valor deve ser o mesmo que InputTensor.DimensionCount.

Repeats

Tipo: const UINT*

Cada valor nessa matriz corresponde a uma das dimensões do tensor de entrada (em ordem). Cada valor é o número de cópias lado a lado para fazer dessa dimensão. Os valores devem ser maiores que 0.

Exemplos

RepeatsCount: 4
Repeats: {1, 1, 3, 3}

InputTensor: (Sizes:{1, 1, 2, 3}, DataType:FLOAT32)
[[[[1, 2, 3]
   [4, 5, 6]]]]

InputTensor: (Sizes:{1, 1, 6, 9}, DataType:FLOAT32)
[[[[1, 2, 3, 1, 2, 3, 1, 2, 3]
   [4, 5, 6, 4, 5, 6, 4, 5, 6] 
   [1, 2, 3, 1, 2, 3, 1, 2, 3] 
   [4, 5, 6, 4, 5, 6, 4, 5, 6] 
   [1, 2, 3, 1, 2, 3, 1, 2, 3] 
   [4, 5, 6, 4, 5, 6, 4, 5, 6]]]]

Disponibilidade

Esse operador foi introduzido no DML_FEATURE_LEVEL_1_0.

Restrições do Tensor

InputTensor e OutputTensor devem ter o mesmo DataType.

Suporte ao Tensor

DML_FEATURE_LEVEL_4_1 e superior

Tensor	Tipo	Contagens de dimensões com suporte	Tipos de dados com suporte
InputTensor	Entrada	1 a 8	FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor	Saída	1 a 8	FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_1 e superior

Tensor	Tipo	Contagens de dimensões com suporte	Tipos de dados com suporte
InputTensor	Entrada	1 a 8	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor	Saída	1 a 8	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_2_1 e superior

Tensor	Tipo	Contagens de dimensões com suporte	Tipos de dados com suporte
InputTensor	Entrada	4	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor	Saída	4	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_1_0 e superior

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


Cabeçalho	directml.h

Compartilhar via