DML_GRU_OPERATOR_DESC 结构 (directml.h)

(输入上的 GRU) 函数) 单层封闭循环单元执行 (标准层。 此运算符使用多个入口来执行此层。 这些门在序列长度维度和 SequenceLengthsTensor 规定的循环中执行多次。

前进方向的公式

向后方向的公式

公式图例

语法

struct DML_GRU_OPERATOR_DESC {
  const DML_TENSOR_DESC           *InputTensor;
  const DML_TENSOR_DESC           *WeightTensor;
  const DML_TENSOR_DESC           *RecurrenceTensor;
  const DML_TENSOR_DESC           *BiasTensor;
  const DML_TENSOR_DESC           *HiddenInitTensor;
  const DML_TENSOR_DESC           *SequenceLengthsTensor;
  const DML_TENSOR_DESC           *OutputSequenceTensor;
  const DML_TENSOR_DESC           *OutputSingleTensor;
  UINT                            ActivationDescCount;
  const DML_OPERATOR_DESC         *ActivationDescs;
  DML_RECURRENT_NETWORK_DIRECTION Direction;
  BOOL                            LinearBeforeReset;
};

成员

InputTensor

类型： const DML_TENSOR_DESC*

包含输入数据的张量 X. 打包 (，并可能将) 填充为一个 大小 为 的 { 1, seq_length, batch_size, input_size }4D 张量。 seq_length是映射到索引的维度 t。 张量不支持 DML_TENSOR_FLAG_OWNED_BY_DML 标志。

WeightTensor

类型： const DML_TENSOR_DESC*

如果双向) ，则包含权重数据的张量，W_[zrh] 和 W_B[zrh] 的 W. 串联 (。 张量具有 大小{ 1, num_directions, 3 * hidden_size, input_size }。 张量不支持 DML_TENSOR_FLAG_OWNED_BY_DML 标志。

RecurrenceTensor

类型： const DML_TENSOR_DESC*

包含重复周期数据的张量，如果双向) ，则R_[zrh] 和 R_B[zrh] 的 R. 串联 (。 张量具有 大小{ 1, num_directions, 3 * hidden_size, hidden_size }。 张量不支持 DML_TENSOR_FLAG_OWNED_BY_DML 标志。

BiasTensor

类型：_Maybenull_ const DML_TENSOR_DESC*

包含偏差数据的可选张量，如果双向) ，则包含 (W_b[zrh]、R_b[zrh]) 和 (W_Bb[zrh]、R_Bb[zrh]) (。 张量具有 大小{ 1, 1, num_directions, 6 * hidden_size }。 张量不支持 DML_TENSOR_FLAG_OWNED_BY_DML 标志。

HiddenInitTensor

类型：_Maybenull_ const DML_TENSOR_DESC*

包含隐藏节点初始值设定项张量H_t-1 的可选张量，用于第一个循环索引 t。 如果未指定，则默认为 0。 此张量具有 大小{ 1, num_directions, batch_size, hidden_size }。 张量不支持 DML_TENSOR_FLAG_OWNED_BY_DML 标志。

SequenceLengthsTensor

类型：_Maybenull_ const DML_TENSOR_DESC*

包含批处理中每个元素的独立seq_length的可选张量。 如果未指定，则批中的所有序列的长度seq_length。 此张量具有 大小{ 1, 1, 1, batch_size }。 张量不支持 DML_TENSOR_FLAG_OWNED_BY_DML 标志。

OutputSequenceTensor

类型：_Maybenull_ const DML_TENSOR_DESC*

一个可选的张量，用于编写隐藏节点的所有中间输出值的串联，H_t。 此张量具有 大小{ seq_length, num_directions, batch_size, hidden_size }。 seq_length映射到循环索引 t。

OutputSingleTensor

类型：_Maybenull_ const DML_TENSOR_DESC*

一个可选的张量，用于写入隐藏节点的最后一个输出值，H_t。 此张量具有 大小{ 1, num_directions, batch_size, hidden_size }。

ActivationDescCount

类型： UINT

此字段确定 ActivationDescs 数组的大小。

ActivationDescs

类型：_Field_size_ (ActivationDescCount) const DML_OPERATOR_DESC*

包含激活运算符的说明 的DML_OPERATOR_DESC 数组，f () 和 g () 。 f () 和 g () 都是独立于方向定义的，这意味着如果在 Direction 中提供DML_RECURRENT_NETWORK_DIRECTION_FORWARD或DML_RECURRENT_NETWORK_DIRECTION_BACKWARD，则必须提供两次激活。 如果提供了 DML_RECURRENT_NETWORK_DIRECTION_BIDIRECTIONAL ，则必须提供四次激活。 对于双向激活，必须为前向提供激活 f () 和 g () ，后跟 f () 和 g () 。

Direction

类型： const DML_RECURRENT_NETWORK_DIRECTION*

运算符的方向 - 向前、向后或双向。

LinearBeforeReset

类型： BOOL

如果为 TRUE ，则指定在计算隐藏门的输出时，应在将线性转换乘以重置门的输出之前应用。 否则为 FALSE。

可用性

此运算符是在 中引入的 DML_FEATURE_LEVEL_1_0。

张量约束

BiasTensor、 HiddenInitTensor、 InputTensor、 OutputSequenceTensor、 OutputSingleTensor、 RecurrenceTensor 和 WeightTensor 必须具有相同 的数据类型。

Tensor 支持

张	种类	支持的维度计数	支持的数据类型
InputTensor	输入	4	FLOAT32、FLOAT16
WeightTensor	输入	4	FLOAT32、FLOAT16
RecurrenceTensor	输入	4	FLOAT32、FLOAT16
BiasTensor	可选输入	4	FLOAT32、FLOAT16
HiddenInitTensor	可选输入	4	FLOAT32、FLOAT16
SequenceLengthsTensor	可选输入	4	UINT32
OutputSequenceTensor	可选输出	4	FLOAT32、FLOAT16
OutputSingleTensor	可选输出	4	FLOAT32、FLOAT16

要求

要求	值
Header	directml.h