estructura DML_TILE_OPERATOR_DESC (directml.h)
Construye un tensor de salida mediante la colocación en mosaico del tensor de entrada. Los elementos de cada dimensión del tensor de entrada se repiten mediante un múltiplo de la matriz Repeats .
Sintaxis
struct DML_TILE_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
UINT RepeatsCount;
const UINT *Repeats;
};
Miembros
InputTensor
Tipo: const DML_TENSOR_DESC*
Tensor del que se va a leer, que contiene los elementos que se van a crear en mosaico.
OutputTensor
Tipo: const DML_TENSOR_DESC*
Tensor en el que se va a escribir, que contendrá la salida en mosaico. Para cada dimensión i
de [0, InputTensor.DimensionCount-1]
, el tamaño de salida se calcula como OutputTensor.Sizes[i] = InputTensor.Sizes[i] * Repeats[i]
. Este tensor debe tener el mismo DimensionCount que el tensor de entrada.
RepeatsCount
Tipo: UINT
Este campo determina el tamaño de la matriz Repeats . Este valor debe ser el mismo que InputTensor.DimensionCount.
Repeats
Tipo: const UINT*
Cada valor de esta matriz corresponde a una de las dimensiones del tensor de entrada (en orden). Cada valor es el número de copias en mosaico para crear esa dimensión. Los valores deben ser mayores que 0.
Ejemplos
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]]]]
Disponibilidad
Este operador se introdujo en DML_FEATURE_LEVEL_1_0
.
Restricciones tensor
InputTensor y OutputTensor deben tener el mismo tipo de datos.
Compatibilidad con Tensor
DML_FEATURE_LEVEL_4_1 y versiones posteriores
Tensor | Kind | Recuentos de dimensiones admitidos | Tipos de datos admitidos |
---|---|---|---|
InputTensor | Entrada | De 1 a 8 | FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8 |
OutputTensor | Resultados | De 1 a 8 | FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8 |
DML_FEATURE_LEVEL_3_1 y versiones posteriores
Tensor | Kind | Recuentos de dimensiones admitidos | Tipos de datos admitidos |
---|---|---|---|
InputTensor | Entrada | De 1 a 8 | FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8 |
OutputTensor | Resultados | De 1 a 8 | FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8 |
DML_FEATURE_LEVEL_2_1 y versiones posteriores
Tensor | Kind | Recuentos de dimensiones admitidos | Tipos de datos admitidos |
---|---|---|---|
InputTensor | Entrada | 4 | FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8 |
OutputTensor | Resultados | 4 | FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8 |
DML_FEATURE_LEVEL_1_0 y versiones posteriores
Tensor | Kind | Recuentos de dimensiones admitidos | Tipos de datos admitidos |
---|---|---|---|
InputTensor | Entrada | 4 | FLOAT32, FLOAT16 |
OutputTensor | Resultados | 4 | FLOAT32, FLOAT16 |
Requisitos
Encabezado | directml.h |