структура DML_FILL_VALUE_CONSTANT_OPERATOR_DESC (directml.h)

Статья
08/23/2023

Заполняет тензор заданным константой Value. Этот оператор выполняет следующий псевдокод.

for each coordinate in OutputTensor
    OutputTensor[coordinate] = Value
endfor

Синтаксис

struct DML_FILL_VALUE_CONSTANT_OPERATOR_DESC {
  const DML_TENSOR_DESC *OutputTensor;
  DML_TENSOR_DATA_TYPE  ValueDataType;
  DML_SCALAR_UNION      Value;
};

Члены

OutputTensor

Тип: const DML_TENSOR_DESC*

Тензор, в который записываются результаты. Этот тензор может иметь любой размер.

ValueDataType

Тип: DML_TENSOR_DATA_TYPE

Тип данных поля Value , который должен соответствовать OutputTensor.DataType.

Value

Тип: DML_SCALAR_UNION

Постоянное значение для заполнения выходных данных с параметром ValueDataType , определяющим способ интерпретации поля.

Примеры

Value = 13.0

OutputTensor: (Sizes:{1,1,2,4}, DataType:FLOAT32)
    [[[[13.0f, 13.0f, 13.0f, 13.0f],
       [13.0f, 13.0f, 13.0f, 13.0f]]]]

Доступность

Этот оператор появился в DML_FEATURE_LEVEL_2_1.

Поддержка тензоров

DML_FEATURE_LEVEL_4_1 и выше

Тензор	Kind	Поддерживаемые счетчики измерений	Поддерживаемые типы данных
OutputTensor	Выходные данные	От 1 до 8	FLOAT64, FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_4_0 и выше

Тензор	Kind	Поддерживаемые счетчики измерений	Поддерживаемые типы данных
OutputTensor	Выходные данные	От 1 до 8	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_2_1 и выше

Тензор	Kind	Поддерживаемые счетчики измерений	Поддерживаемые типы данных
OutputTensor	Выходные данные	4	FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

Требования

Требование	Значение
Минимальная версия клиента	сборка Windows 10 20348
Минимальная версия сервера	сборка Windows 10 20348
Верхняя часть	directml.h

Поделиться через