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

Статья
02/29/2024

Приводит каждый элемент во входе к типу данных выходного тензора и сохраняет результат в соответствующем элементе выходных данных.

Синтаксис

struct DML_CAST_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

Тензор, в который записываются результаты. Размеры этого тензора должны совпадать с InputTensor.

OutputTensor

Тип: const DML_TENSOR_DESC*

Указатель на константную DML_TENSOR_DESC , содержащую описание тензора для записи результатов.

Некоторые типы данных могут не поддерживаться на определенном оборудовании. Чтобы определить, поддерживается ли тип данных, используйте IDMLDevice::CheckFeatureSupport с DML_FEATURE_TENSOR_DATA_TYPE_SUPPORT.

Доступность

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

Ограничения тензоров

InputTensor и OutputTensor должны иметь одинаковые значения DimensionCount и Sizes.

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

DML_FEATURE_LEVEL_4_1 и выше

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

DML_FEATURE_LEVEL_3_1 и выше

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

DML_FEATURE_LEVEL_3_0 и выше

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

DML_FEATURE_LEVEL_1_0 и выше

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

Требования

Требование	Значение
Заголовок	directml.h

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