struttura DML_TILE_OPERATOR_DESC (directml.h)

Articolo
08/23/2023

Costruisce un tensore di output affiancando il tensore di input. Gli elementi in ogni dimensione del tensore di input vengono ripetuti da un multiplo nella matrice Repeats .

Sintassi

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

Members

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensore da cui leggere, che contiene gli elementi da riquadri.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Tensore in cui scrivere, che conterrà l'output affiancato. Per ogni dimensione i in [0, InputTensor.DimensionCount-1], le dimensioni di output vengono calcolate come OutputTensor.Sizes[i] = InputTensor.Sizes[i] * Repeats[i]. Questo tensore deve avere lo stesso Valore DimensionCount del tensore di input.

RepeatsCount

Tipo: UINT

Questo campo determina le dimensioni della matrice Repeats . Questo valore deve essere uguale a InputTensor.DimensionCount.

Repeats

Tipo: const UINT*

Ogni valore in questa matrice corrisponde a una delle dimensioni del tensore di input (in ordine). Ogni valore è il numero di copie affiancate da creare di tale dimensione. I valori devono essere maggiori di 0.

Esempio

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]]]]

Disponibilità

Questo operatore è stato introdotto in DML_FEATURE_LEVEL_1_0.

Vincoli tensor

InputTensor e OutputTensor devono avere lo stesso Tipo di dati.

Supporto di Tensor

DML_FEATURE_LEVEL_4_1 e versioni successive

Tensore	Tipo	Conteggi delle dimensioni supportati	Tipi di dati supportati
InputTensor	Input	da 1 a 8	FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputTensor	Output	da 1 a 8	FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_1 e versioni successive

Tensore	Tipo	Conteggi delle dimensioni supportati	Tipi di dati supportati
InputTensor	Input	da 1 a 8	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor	Output	da 1 a 8	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_2_1 e versioni successive

Tensore	Tipo	Conteggi delle dimensioni supportati	Tipi di dati supportati
InputTensor	Input	4	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor	Output	4	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_1_0 e versioni successive

Tensore	Tipo	Conteggi delle dimensioni supportati	Tipi di dati supportati
InputTensor	Input	4	FLOAT32, FLOAT16
OutputTensor	Output	4	FLOAT32, FLOAT16

Requisiti


Intestazione	directml.h

Condividi tramite