DirectML 结构
以下结构在 DirectML.h
中声明。
本节内容
主题和说明 |
---|
DML_ACTIVATION_CELU_OPERATOR_DESC。 对 InputTensor 中的每个元素执行连续可微指数线性单元 (CELU) 激活函数,并将结果放入 OutputTensor 的相应元素中。 |
DML_ACTIVATION_ELU_OPERATOR_DESC。 描述对输入中的每个元素执行指数线性单元 (ELU) 激活函数的 DirectML 运算符。 |
DML_ACTIVATION_GELU_OPERATOR_DESC。 对 InputTensor 中的每个元素执行高斯误差线性单元 (GELU) 激活函数,将结果放入 OutputTensor 的相应元素中。 |
DML_ACTIVATION_HARD_SWISH_OPERATOR_DESC。 对 InputTensor 中的每个元素执行 hard swish 激活函数,将结果放入 OutputTensor 的相应元素中。 |
DML_ACTIVATION_HARDMAX_OPERATOR_DESC。 描述对输入执行 hardmax 函数的 DirectML 激活运算符。 |
DML_ACTIVATION_HARDMAX1_OPERATOR_DESC。 对 InputTensor 的每个元素执行 hardmax 函数,将结果放入 OutputTensor 的相应元素中。 |
DML_ACTIVATION_HARD_SIGMOID_OPERATOR_DESC。 描述对输入中的每个元素执行 hard sigmoid 函数的 DirectML 激活运算符。 |
DML_ACTIVATION_IDENTITY_OPERATOR_DESC。 描述执行 identity 函数的 DirectML 激活运算符。 |
DML_ACTIVATION_LEAKY_RELU_OPERATOR_DESC。 描述对输入中的每个元素执行泄漏修正线性单元 (ReLU) 激活函数的 DirectML 运算符。 |
DML_ACTIVATION_LINEAR_OPERATOR_DESC。 描述对输入中的每个元素执行线性激活函数的 DirectML 运算符。 |
DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC。 描述在对输入执行 log-of-softmax 激活函数的 DirectML 运算符。 |
DML_ACTIVATION_LOG_SOFTMAX1_OPERATOR_DESC。 对 InputTensor 的每个元素执行自然的 log-of-softmax 激活函数,将结果放入 OutputTensor 的相应元素中。 |
DML_ACTIVATION_PARAMETERIZED_RELU_OPERATOR_DESC。 描述对输入中的每个元素执行参数化修正线性单元 (ReLU) 激活函数的 DirectML 运算符。 |
DML_ACTIVATION_PARAMETRIC_SOFTPLUS_OPERATOR_DESC。 描述对输入中的每个元素执行参数 softplus 激活函数的 DirectML 运算符。 |
DML_ACTIVATION_RELU_GRAD_OPERATOR_DESC。 计算计算修正线性单元 (ReLU) 的反向传播梯度。 |
DML_ACTIVATION_RELU_OPERATOR_DESC。 描述对输入中的每个元素执行修正线性单元 (ReLU) 激活函数的 DirectML 运算符。 |
DML_ACTIVATION_SCALED_ELU_OPERATOR_DESC。 描述对输入中的每个元素执行缩放指数线性单元 (ELU) 激活函数的 DirectML 运算符。 |
DML_ACTIVATION_SCALED_TANH_OPERATOR_DESC。 描述对输入中的每个元素执行缩放双曲正切激活函数的 DirectML 运算符。 |
DML_ACTIVATION_SHRINK_OPERATOR_DESC。 描述对输入执行按元素收缩激活函数的 DirectML 运算符。 |
DML_ACTIVATION_SIGMOID_OPERATOR_DESC。 描述对输入中的每个元素执行 sigmoid 激活函数的 DirectML 运算符。 |
DML_ACTIVATION_SOFTMAX_OPERATOR_DESC。 描述对输入执行 softmax 激活函数的 DirectML 运算符。 |
DML_ACTIVATION_SOFTMAX1_OPERATOR_DESC。 对 InputTensor 执行 softmax 激活函数,将结果放入 OutputTensor 的相应元素中。 |
DML_ACTIVATION_SOFTPLUS_OPERATOR_DESC。 描述对输入中的每个元素执行 softplus 激活函数的 DirectML 运算符。 |
DML_ACTIVATION_SOFTSIGN_OPERATOR_DESC。 描述对输入中的每个元素执行 softsign 激活函数的 DirectML 运算符。 |
DML_ACTIVATION_SWISH_OPERATOR_DESC。 对 InputTensor 中的每个元素执行 swish 激活函数,将结果放入 OutputTensor 的相应元素中。 |
DML_ACTIVATION_TANH_OPERATOR_DESC。 描述对输入中的每个元素执行双曲正切激活函数的 DirectML 运算符。 |
DML_ACTIVATION_THRESHOLDED_RELU_OPERATOR_DESC。 描述对输入中的每个元素执行阈值修正线性单元 (ReLU) 激活函数的 DirectML 运算符。 |
DML_ADAM_OPTIMIZER_OPERATOR_DESC。 根据 Adam(ADAptive Moment 估计)算法,使用提供的梯度计算更新的权重(参数)。 此运算符是优化器,通常用于在训练循环的权重更新步骤中执行梯度下降。 |
DML_AVERAGE_POOLING_OPERATOR_DESC。 描述对输入执行平均池化函数的 DirectML 运算符。 |
DML_AVERAGE_POOLING1_OPERATOR_DESC。 对输入张量上滑动窗口中的元素求平均值。 |
DML_ARGMAX_OPERATOR_DESC。 输出输入张量一个或多个维度内最大值元素的索引。 |
DML_ARGMIN_OPERATOR_DESC。 输出输入张量一个或多个维度内最小值元素的索引。 |
DML_AVERAGE_POOLING_GRAD_OPERATOR_DESC。 计算平均池化的反向传播梯度(请参阅 DML_AVERAGE_POOLING_OPERATOR_DESC)。 |
DML_BATCH_NORMALIZATION_GRAD_OPERATOR_DESC。 计算批量规范化的反向传播梯度。 |
DML_BATCH_NORMALIZATION_OPERATOR_DESC。 描述对输入执行批规范化函数的 DirectML 运算符。 |
DML_BATCH_NORMALIZATION_TRAINING_GRAD_OPERATOR_DESC。 计算批量规范化训练的反向传播梯度。 |
DML_BATCH_NORMALIZATION_TRAINING_OPERATOR_DESC。 对输入执行批规范化。 |
DML_BINDING_DESC。 包含绑定的说明,以便可以通过调用其中一个 IDMLBindingTable 方法将其添加到绑定表。 |
DML_BINDING_PROPERTIES。 包含有关特定已编译运算符或运算符初始值设定项的绑定要求的信息。 |
DML_BINDING_TABLE_DESC。 指定 IDMLDevice::CreateBindingTable 和 IDMLBindingTable::Reset 的参数。 |
DML_BUFFER_ARRAY_BINDING。 指定是单个缓冲区绑定数组的资源绑定。 |
DML_BUFFER_BINDING。 指定 Direct3D 12 缓冲区中的字节范围描述的资源绑定,其通过 ID3D12Resource 的偏移量和大小表示。 |
DML_BUFFER_TENSOR_DESC。 描述将存储在 Direct3D 12 缓冲区资源中的张量。 |
DML_CAST_OPERATOR_DESC。 描述执行 cast 函数 f(x) = cast(x),将输入中的每个元素强制转换为输出张量的数据类型,并将结果存储在输出的相应元素中的 DirectML 数据重新组织运算符。 |
DML_CONVOLUTION_INTEGER_OPERATOR_DESC。 使用 InputTensor 执行 FilterTensor 的卷积。 此运算符对整数数据执行正向卷积。 |
DML_CONVOLUTION_OPERATOR_DESC。 描述对输入执行卷积函数的 DirectML 矩阵乘法运算符。 |
DML_CUMULATIVE_PRODUCT_OPERATOR_DESC。 沿轴将张量的元素相乘,将乘积的运行计数写入输出张量。 |
DML_CUMULATIVE_SUMMATION_OPERATOR_DESC。 沿轴将张量的元素相加,将总和的运行计数写入输出张量。 |
DML_DEPTH_TO_SPACE_OPERATOR_DESC。 描述将数据从深度重新排列(置换)为空间数据块的 DirectML 数据重新组织运算符。 |
DML_DEPTH_TO_SPACE1_OPERATOR_DESC。 将数据从深度重新排列(置换)为空间数据块。 该运算符输出输入张量的副本,其中深度维度中的值在空间块中移动到高度和宽度维度。 |
DML_DIAGONAL_MATRIX_OPERATOR_DESC。 描述生成一个类恒等矩阵,其中主对角线上为 1,其他位置均为 0 的 DirectML 数学运算符。 |
DML_DIAGONAL_MATRIX1_OPERATOR_DESC。 生成一个类恒等矩阵,其中沿着给定的对角线跨度值为 1(或其他显式值),其他元素填充为输入值或 0(如果没有传递 InputTensor)。 |
DML_DYNAMIC_QUANTIZE_LINEAR_OPERATOR_DESC。 计算量化 InputTensor 所需的量化比例和零点值,然后应用该量化,将结果写入 OutputTensor。 |
DML_ELEMENT_WISE_ABS_OPERATOR_DESC。 描述执行按元素绝对值函数 f(x) = abs(x * scale + bias)(其中 scale 和 bias 项是可选的)的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_ACOS_OPERATOR_DESC。 描述执行按元素反余弦函数 f(x) = acos(x * scale + bias)(其中 scale 和 bias 项是可选的)的 DirectML 三角运算符。 |
DML_ELEMENT_WISE_ACOSH_OPERATOR_DESC。 描述执行按元素反双曲余弦函数 f(x) = log(x + sqrt(x * x - 1)) * scale + bias(其中 scale 和 bias 项是可选的)的 DirectML 三角运算符。 |
DML_ELEMENT_WISE_ADD_OPERATOR_DESC。 描述执行将 ATensor 中的每个元素添加到 BTensor 中其相应元素的函数的 DirectML 运算符。 |
DML_ELEMENT_WISE_ADD1_OPERATOR_DESC。 描述执行将 ATensor 中的每个元素添加到 BTensor 中其相应元素的函数(即 f(a, b) = a + b),并可选择融合激活的 DirectML 运算符。 |
DML_ELEMENT_WISE_ASIN_OPERATOR_DESC。 描述执行按元素反正弦函数 f(x) = asin(x * scale + bias)(其中 scale 和 bias 项是可选的)的 DirectML 三角运算符。 |
DML_ELEMENT_WISE_ASINH_OPERATOR_DESC。 描述执行按元素反双曲正弦函数 f(x) = log(x + sqrt(x * x + 1))(其中 scale 和 bias 项是可选的)的 DirectML 三角运算符。 |
DML_ELEMENT_WISE_ATAN_OPERATOR_DESC。 描述执行按元素反正切函数 f(x) = atan(x * scale + bias)(其中 scale 和 bias 项是可选的)的 DirectML 三角运算符。 |
DML_ELEMENT_WISE_ATANH_OPERATOR_DESC。 描述执行按元素反双曲正切函数 f(x) = (log((1 + x) / (1 - x)) / 2) * scale + bias(其中 scale 和 bias 项是可选的)的 DirectML 三角运算符。 |
DML_ELEMENT_WISE_ATAN_YX_OPERATOR_DESC。 计算 ATensor 和 BTensor 每个元素的 2 参数反正切值(其中 ATensor 是 Y 轴,BTensor 是 X 轴),并将结果放入 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_BIT_AND_OPERATOR_DESC。 计算输入张量的每个相应元素之间的按位 AND 运算,并将结果写入输出张量。 |
DML_ELEMENT_WISE_BIT_COUNT_OPERATOR_DESC。 计算输入张量每个元素的按位 NOT 运算,并将结果写入输出张量。 |
DML_ELEMENT_WISE_BIT_NOT_OPERATOR_DESC。 计算输入张量每个元素的按位总体计数(位数设置为 1),并将结果写入输出张量中。 |
DML_ELEMENT_WISE_BIT_OR_OPERATOR_DESC。 计算输入张量的每个相应元素之间的按位 OR 运算,并将结果写入输出张量。 |
DML_ELEMENT_WISE_BIT_SHIFT_LEFT_OPERATOR_DESC。 对 ATensor 的每个元素执行逻辑左移 BTensor 的相应元素给出的位数,并将结果放入 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_BIT_SHIFT_RIGHT_OPERATOR_DESC。 对 ATensor 的每个元素执行逻辑右移 BTensor 的相应元素给出的位数,并将结果放入 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_BIT_XOR_OPERATOR_DESC。 计算输入张量的每个相应元素之间的按位 XOR (eXclusive OR) 运算,并将结果写入输出张量。 |
DML_ELEMENT_WISE_CEIL_OPERATOR_DESC。 描述执行按元素向上取整函数 f(x) = ceil(x * scale + bias)(其中 scale 和 bias 项是可选的)的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_CLIP_GRAD_OPERATOR_DESC。 计算按元素剪辑的反向传播梯度。 |
DML_ELEMENT_WISE_CLIP_GRAD1_OPERATOR_DESC。 计算 DML_ELEMENT_WISE_CLIP1_OPERATOR_DESC 的反向传播梯度。 |
DML_ELEMENT_WISE_CLIP_OPERATOR_DESC。 描述执行按元素剪辑函数 f(x) = clamp(x * scale + bias, minValue, maxValue)(其中 scale 和 bias 项是可选的,而 clamp(x) = min(maxValue, max(minValue, x))的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_CLIP1_OPERATOR_DESC。 对 InputTensor 的每个元素执行箝位(或限制)运算,将结果放入 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_CONSTANT_POW_OPERATOR_DESC。 描述执行按元素恒幂函数 f(x) = pow(x * scale + bias, exponent)(其中 scale 和 bias 项是可选的)的 DirectML 运算符。 |
DML_ELEMENT_WISE_COS_OPERATOR_DESC。 描述执行按元素余弦函数 f(x) = cos(x * scale + bias)(其中 scale 和 bias 项是可选的)的 DirectML 三角运算符。 |
DML_ELEMENT_WISE_COSH_OPERATOR_DESC。 描述执行按元素双曲余弦函数 f(x) = ((e^x + e^-x) / 2) * scale + bias(其中 scale 和 bias 项是可选的)的 DirectML 三角运算符。 |
DML_ELEMENT_WISE_DEQUANTIZE_LINEAR_OPERATOR_DESC。 描述对 InputTensor 中每个元素相对于 ScaleTensor 和 ZeroPointTensor 中其相应元素执行线性反量化函数的 DirectML 运算符。 |
DML_ELEMENT_WISE_DIFFERENCE_SQUARE_OPERATOR_DESC。 从 ATensor 的相应元素中减去 BTensor 的每个元素,将结果与其自身相乘,然后将结果放入 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_DIVIDE_OPERATOR_DESC。 描述执行将 ATensor 中的每个元素除以 BTensor 中其相应元素的函数的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_ERF_OPERATOR_DESC。 描述执行按元素自然指数函数 f(x) = exp(x * scale + bias)(其中 scale 和 bias 项是可选的)的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_EXP_OPERATOR_DESC。 描述执行按元素自然指数函数 f(x) = exp(x * scale + bias)(其中 scale 和 bias 项是可选的)的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_FLOOR_OPERATOR_DESC。 描述执行按元素向下取整函数 f(x) = floor(x * scale + bias)(其中 scale 和 bias 项是可选的)的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_IDENTITY_OPERATOR_DESC。 描述执行按元素 identity 函数 f(x) = x * scale + bias 的 DirectML 泛型运算符。。 |
DML_ELEMENT_WISE_IF_OPERATOR_DESC。 描述实质上执行三元 if 语句的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_IS_INFINITY_OPERATOR_DESC。 根据给定的 InfinityMode,检查 InputTensor 的每个元素是否符合 IEEE-754 -inf、inf 或两者,并将结果(1 表示 true,0 表示 false)放入 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_IS_NAN_OPERATOR_DESC。 描述按元素确定输入是否为 NaN 的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_LOGICAL_AND_OPERATOR_DESC。 描述对 ATensor 中的每个元素及其 BTensor 中相应元素之间执行逻辑 AND 函数的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_LOGICAL_EQUALS_OPERATOR_DESC。 描述对 ATensor 中的每个元素及其 BTensor 中相应元素之间执行逻辑相等性函数的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_LOGICAL_GREATER_THAN_OPERATOR_DESC。 描述对 ATensor 中的每个元素及其 BTensor 中相应元素之间执行逻辑大于函数的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_LOGICAL_GREATER_THAN_OR_EQUAL_OPERATOR_DESC。 对输入张量的每对相应元素执行逻辑大于或等于,将结果(1 表示 true,0 表示 false)放入 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_LOGICAL_LESS_THAN_OPERATOR_DESC。 描述对 ATensor 中的每个元素及其 BTensor 中相应元素之间执行逻辑小于函数的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_LOGICAL_LESS_THAN_OR_EQUAL_OPERATOR_DESC。 对输入张量的每对相应元素执行逻辑小于或等于,将结果(1 表示 true,0 表示 false)放入 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_LOGICAL_NOT_OPERATOR_DESC。 描述对输入中的每个元素执行逻辑 NOT 函数的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_LOGICAL_OR_OPERATOR_DESC。 描述对 ATensor 中的每个元素及其 BTensor 中相应元素之间执行逻辑 OR 函数的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_LOGICAL_XOR_OPERATOR_DESC。 描述对 ATensor 中每个元素及其 BTensor 中相应元素之间执行逻辑独占 OR (XOR) 函数的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_LOG_OPERATOR_DESC。 描述执行按元素自然对数函数 f(x) = log(x * scale + bias)(其中 scale 和 bias 项是可选的)的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_MAX_OPERATOR_DESC。 描述对 ATensor 中的每个元素及其 BTensor 中相应元素之间执行最大值函数的 DirectML 数学归约运算符。 |
DML_ELEMENT_WISE_MEAN_OPERATOR_DESC。 描述对 ATensor 中的每个元素及其 BTensor 中相应元素之间执行算术平均值函数的 DirectML 数学归约运算符。 |
DML_ELEMENT_WISE_MIN_OPERATOR_DESC。 描述对 ATensor 中的每个元素及其 BTensor 中相应元素之间执行最小值函数的 DirectML 数学归约运算符。 |
DML_ELEMENT_WISE_MODULUS_FLOOR_OPERATOR_DESC。 对输入张量中的每对相应元素计算模数(结果与 Python 模数相同),将结果放入 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_MODULUS_TRUNCATE_OPERATOR_DESC。 对输入张量中的每对相应元素计算 C 模数运算符,并将结果放入 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_MULTIPLY_OPERATOR_DESC。 描述执行将 ATensor 中的每个元素乘以 BTensor 中其相应元素的函数的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_NEGATE_OPERATOR_DESC。 否定 InputTensor 的每个元素,将结果存储到 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_POW_OPERATOR_DESC。 描述执行按元素幂函数 f(x, exponent) = pow(x * scale + bias, exponent)(其中 scale 和 bias 项是可选的)的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_QUANTIZE_LINEAR_OPERATOR_DESC。 描述对 InputTensor 中每个元素相对于 ScaleTensor 和 ZeroPointTensor 中其相应元素执行线性量化函数的 DirectML 运算符。 |
DML_ELEMENT_WISE_QUANTIZED_LINEAR_ADD_OPERATOR_DESC。 将 ATensor 中的每个元素与 BTensor 中其相应元素相加,将结果放入 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_RECIP_OPERATOR_DESC。 描述对输入中的每个元素执行倒数函数的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_ROUND_OPERATOR_DESC。 将 InputTensor 的每个元素舍入到整数值,将结果放入 OutputTensor 的相应元素中。 |
DML_ELEMENT_WISE_SIGN_OPERATOR_DESC。 描述对输入执行按元素收缩激活函数的 DirectML 运算符。 |
DML_ELEMENT_WISE_SIN_OPERATOR_DESC。 描述执行按元素正弦函数 f(x) = sin(x * scale + bias)(其中 scale 和 bias 项是可选的)的 DirectML 三角运算符。 |
DML_ELEMENT_WISE_SINH_OPERATOR_DESC。 描述执行按元素双曲正弦函数 f(x) = ((e^x - e^-x) / 2) * scale + bias(其中 scale 和 bias 项是可选的)的 DirectML 三角运算符。 |
DML_ELEMENT_WISE_SQRT_OPERATOR_DESC。 描述对输入中的每个元素执行平方根函数的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_SUBTRACT_OPERATOR_DESC。 描述执行从 ATensor 中其相应元素减去 BTensor 中的每个元素的函数的 DirectML 数学运算符。 |
DML_ELEMENT_WISE_TAN_OPERATOR_DESC。 描述执行按元素正切函数 f(x) = tan(x * scale + bias)(其中 scale 和 bias 项是可选的)的 DirectML 三角运算符。 |
DML_ELEMENT_WISE_TANH_OPERATOR_DESC。 描述执行按元素反双曲正切函数 f(x) = tanh(x) * scale + bias(其中 scale 和 bias 项是可选的)的 DirectML 三角运算符。 |
DML_ELEMENT_WISE_THRESHOLD_OPERATOR_DESC。 描述执行按元素阈值函数 f(x) = max(x * scale + bias, min),(其中 scale 和 bias 项是可选的)的 DirectML 数学运算符。 |
DML_FEATURE_DATA_TENSOR_DATA_TYPE_SUPPORT。 提供有关 DirectML 设备是否支持张量内特定数据类型的详细信息。 |
DML_FEATURE_QUERY_TENSOR_DATA_TYPE_SUPPORT。 用于查询 DirectML 设备是否支持张量内的特定数据类型。 |
DML_FILL_VALUE_CONSTANT_OPERATOR_DESC。 用给定的常量值填充张量。 |
DML_FILL_VALUE_SEQUENCE_OPERATOR_DESC。 用序列填充张量。 |
DML_GATHER_ELEMENTS_OPERATOR_DESC。 使用索引张量沿给定轴从输入张量收集元素以重新映射到输入。 |
DML_GATHER_ND_OPERATOR_DESC。 收集输入张量中的元素,使用索引张量将索引重新映射到输入的整个子块。 |
DML_GATHER_ND1_OPERATOR_DESC。 收集输入张量中的元素,使用索引张量将索引重新映射到输入的整个子块。 |
DML_GATHER_OPERATOR_DESC。 描述 DirectML 数据重新组织运算符,当给定排名 r >= 1 的数据张量和排名 q 的索引张量时,收集数据轴维度的条目(默认情况下,最外层为 axis == 0),并在排名 q + (r - 1) 的输出张量中连接它们。 |
DML_GEMM_OPERATOR_DESC。 描述对输入执行常规矩阵乘法函数 (y = alpha * transposeA(A) * transposeB(B) + beta * C) 的 DirectML 运算符。 |
DML_GRAPH_DESC。 描述用于编译组合优化运算符的 DirectML 运算符图。 |
DML_GRAPH_EDGE_DESC。 由 DML_GRAPH_DESC 定义并传递给 IDMLDevice1::CompileGraph 的 DirectML 运算符图形内连接的通用容器。 |
DML_GRAPH_NODE_DESC。 由 DML_GRAPH_DESC 定义并传递给 IDMLDevice1::CompileGraph 的 DirectML 运算符图形内节点的通用容器。 |
DML_GRU_OPERATOR_DESC。 描述对输入执行(标准层)单层门控循环单元 (GRU) 函数的 DirectML 深度学习运算符。 |
DML_INPUT_GRAPH_EDGE_DESC。 描述由 DML_GRAPH_DESC 定义并传递给 IDMLDevice1::CompileGraph 的 DirectML 运算符图形内的连接。 此结构用于定义从图形输入到内部节点输入的连接。 |
DML_INTERMEDIATE_GRAPH_EDGE_DESC。 描述由 DML_GRAPH_DESC 定义并传递给 IDMLDevice1::CompileGraph 的 DirectML 运算符图形内的连接。 此结构用于定义内部节点之间的连接。 |
DML_JOIN_OPERATOR_DESC。 描述对输入张量数组执行联接函数的 DirectML 运算符。 |
DML_LOCAL_RESPONSE_NORMALIZATION_GRAD_OPERATOR_DESC。 计算本地响应规范化的反向传播梯度。 |
DML_LOCAL_RESPONSE_NORMALIZATION_OPERATOR_DESC。 描述对输入执行本地响应规范化 (LRN) 函数的 DirectML 运算符。 |
DML_LP_NORMALIZATION_OPERATOR_DESC。 描述沿输入张量的指定轴执行 Lp 规范化函数的 DirectML 运算符。 |
DML_LP_POOLING_OPERATOR_DESC。 描述跨输入张量执行 Lp 池化函数的 DirectML 运算符。 |
DML_LP_POOLING1_OPERATOR_DESC。 通过输入张量计算滑动窗口中元素的 LP 规范化值。 |
DML_LSTM_OPERATOR_DESC。 描述对输入执行单层长短期内存 (LSTM) 函数的 DirectML 深度学习运算符。 |
DML_MATRIX_MULTIPLY_INTEGER_OPERATOR_DESC。 对整数数据执行矩阵乘法函数。 |
DML_MATRIX_MULTIPLY_INTEGER_TO_FLOAT_OPERATOR_DESC。 对整数输入张量数据执行矩阵乘法函数,并生成浮点输出。 |
DML_MAX_POOLING_GRAD_OPERATOR_DESC。 计算最大最大池化的反向传播梯度(请参阅 DML_MAX_POOLING2_OPERATOR_DESC)。 |
DML_MAX_POOLING_OPERATOR_DESC。 描述跨输入张量执行最大池化函数的 DirectML 运算符。 |
DML_MAX_POOLING1_OPERATOR_DESC。 描述(根据内核大小、步幅大小和垫长)跨输入张量执行最大池化函数,y = max(x1 + x2 + … x_pool_size) 的 DirectML 运算符。 |
DML_MAX_POOLING2_OPERATOR_DESC。 计算输入张量上滑动窗口内元素的最大值,并可选择返回所选最大值的索引。 |
DML_MAX_UNPOOLING_OPERATOR_DESC。 描述用零填充给定形状(显式或输入形状加填充)的输出张量,然后将输入张量中的每个值写入到与相应索引数组元素偏移量处的输出张量的 DirectML 运算符。 |
DML_MEAN_VARIANCE_NORMALIZATION_OPERATOR_DESC。 描述对输入张量执行平均方差规范化函数的 DirectML 运算符。 |
DML_MEAN_VARIANCE_NORMALIZATION1_OPERATOR_DESC。 对输入张量执行平均方差规范化函数。 此运算符将计算输入张量的平均值和方差以执行规范化。 |
DML_MULTIHEAD_ATTENTION_OPERATOR_DESC。 执行多头注意力运算。 |
DML_NONZERO_COORDINATES_OPERATOR_DESC。 计算输入张量所有非零元素的 N 维坐标。 |
DML_ONE_HOT_OPERATOR_DESC。 描述生成一个张量,其中每个元素都填充两个值(即“on”或“off”值)的 DirectML 运算符。 |
DML_OPERATOR_DESC。 运算符说明的通用容器。 使用此结构中指定的参数构造 DirectML 运算符。 |
DML_OPERATOR_GRAPH_NODE_DESC。 描述由 DML_GRAPH_DESC 定义并传递给 IDMLDevice1::CompileGraph 的 DirectML 运算符图形内的节点。 |
DML_OUTPUT_GRAPH_EDGE_DESC。 描述由 DML_GRAPH_DESC 定义并传递给 IDMLDevice1::CompileGraph 的 DirectML 运算符图形内的连接。 此结构用于定义从内部节点的输出到图形输出的连接。 |
DML_PADDING_OPERATOR_DESC。 描述在边缘上用零(或一些其他值)来膨胀输入张量的 DirectML 数据重新组织运算符。 |
DML_PADDING1_OPERATOR_DESC。 使用边缘上的常量或镜像值来膨胀输入张量,并将结果写入输出。 |
DML_QUANTIZED_LINEAR_AVERAGE_POOLING_OPERATOR_DESC。 通过输入张量计算滑动窗口中元素的量化值的平均值。 此运算符在数学上等效于对输入进行去量化、平均池化,然后量化输出。 |
DML_QUANTIZED_LINEAR_CONVOLUTION_OPERATOR_DESC。 使用 InputTensor 执行 FilterTensor 的卷积。 此运算符对量化数据执行正向卷积。 此运算符在数学上等效于对输入进行去量化、卷积,然后量化输出。 |
DML_QUANTIZED_LINEAR_MATRIX_MULTIPLY_OPERATOR_DESC。 对量化数据执行矩阵乘法函数。 此运算符在数学上等效于对输入进行去量化、执行矩阵相乘,然后量化输出。 |
DML_RANDOM_GENERATOR_OPERATOR_DESC。 用确定性生成的、伪随机的、均匀分布的位填充输出张量。 此运算符(可选)还可以输出更新的内部生成器状态,该状态可在运算符的后续执行期间使用。 |
DML_REDUCE_OPERATOR_DESC。 描述对输入执行指定归约函数的 DirectML 运算符。 |
DML_RESAMPLE_GRAD_OPERATOR_DESC。 计算重新取样的反向传播梯度(请参阅 DML_RESAMPLE1_OPERATOR_DESC)。 |
DML_RESAMPLE_GRAD1_OPERATOR_DESC。 计算 DML_RESAMPLE2_OPERATOR_DESC 的反向传播梯度。 |
DML_RESAMPLE_OPERATOR_DESC。 描述使用比例系数计算目标张量大小,将源中的元素重新取样到目标张量的 DirectML 运算符。 |
DML_RESAMPLE1_OPERATOR_DESC。 使用比例因子计算目标张量大小,将源中的元素重新采样到目标张量。 可以使用线性或最接近的邻居内插模式。 |
DML_RESAMPLE2_OPERATOR_DESC。 使用比例因子计算目标张量大小,将源中的元素重新采样到目标张量。 |
DML_REVERSE_SUBSEQUENCES_OPERATOR_DESC。 反转张量一个或多个子序列的元素。 根据提供的轴和序列长度选择要反转的子序列集。 |
DML_RNN_OPERATOR_DESC。 描述对输入执行单层简单递归神经网络 (RNN) 函数的 DirectML 深度学习运算符。 |
DML_ROI_ALIGN_GRAD_OPERATOR_DESC。 计算 ROI_ALIGN 和 ROI_ALIGN1 的反向传播梯度。 |
DML_ROI_ALIGN_OPERATOR_DESC。 执行 ROI 对齐操作,如 Mask R-CNN 论文中所述。 总之,该操作从输入图像张量中提取裁剪,并使用指定的 InterpolationMode 将其大小调整为由 OutputTensor 的最后两个维度指定的通用输出大小。 |
DML_ROI_ALIGN1_OPERATOR_DESC。 执行 ROI 对齐操作,如 Mask R-CNN 论文中所述。 总之,该操作从输入图像张量中提取裁剪后的窗口,并使用指定的 InterpolationMode 将其大小调整为由 OutputTensor 的最后两个维度指定的通用输出大小。 |
DML_ROI_POOLING_OPERATOR_DESC。 描述(根据感兴趣的区域或 ROI)跨输入张量执行池化函数的 DirectML 运算符。 |
DML_SCALAR_UNION。 标量类型的联合。 |
DML_SCALE_BIAS。 包含提供给 DirectML 运算符的 scale 和 bias 项的值。 |
DML_SCATTER_ND_OPERATOR_DESC。 将整个输入张量复制到输出,然后使用更新张量中的相应值覆盖所选索引。 |
DML_SCATTER_OPERATOR_DESC。 描述将整个输入张量复制到输出,然后使用更新张量中的相应值覆盖所选索引的 DirectML 运算符。 |
DML_SIZE_2D。 包含的值可表示张量内元素 2-D 平面的大小(提供给 DirectML 运算符)、2-D 刻度或任意 2-D 宽度/高度值。 |
DML_SLICE_GRAD_OPERATOR_DESC。 计算切片的反向传播梯度(请参阅 DML_SLICE1_OPERATOR_DESC)。 |
DML_SLICE_OPERATOR_DESC。 描述沿多个轴生成输入张量切片的 DirectML 数据重新组织运算符。 |
DML_SLICE1_OPERATOR_DESC。 提取输入张量的单一子区域(“切片”)。 |
DML_SPACE_TO_DEPTH_OPERATOR_DESC。 描述将空间数据块重新排列为深度的 DirectML 数据重新组织运算符。 |
DML_SPACE_TO_DEPTH1_OPERATOR_DESC。 将空间数据块重新排列为深度。 该运算符输出输入张量的副本,其中高度和宽度维度中的值将移动到深度维度。 |
DML_SPLIT_OPERATOR_DESC。 描述沿指定的轴将输入张量拆分为多个输出张量的 DirectML 数据重新组织运算符。 |
DML_TENSOR_DESC。 DirectML 张量说明的通用容器。 |
DML_TILE_OPERATOR_DESC。 描述通过平铺输入张量来构造输出张量的 DirectML 数据重新组织运算符。 |
DML_TOP_K_OPERATOR_DESC。 描述沿指定轴检索前 K 个元素的 DirectML 归约运算符。 |
DML_TOP_K1_OPERATOR_DESC。 沿 InputTensor 的轴选择每个序列中的最大或最小 K 元素,并分别返回 OutputValueTensor 和 OutputIndexTensor 中这些元素的值和索引。 |
DML_UPSAMPLE_2D_OPERATOR_DESC。 描述可向上采样输入张量中所包含图像的 DirectML 图像处理运算符。 |
DML_VALUE_SCALE_2D_OPERATOR_DESC。 描述对输入张量中的值执行按元素缩放和偏差函数的 DirectML 运算符。 |