FusedTransformerEncoderLayer

class paddle.incubate.nn. FusedTransformerEncoderLayer ( d_model, nhead, dim_feedforward, dropout_rate=0.1, activation='relu', attn_dropout_rate=None, act_dropout_rate=None, normalize_before=False, weight_attr=None, bias_attr=None ) [源代码]

FusedTransformer 编码器层由两个子层组成:多头自注意力机制和前馈神经网络。如果 normalize_beforeTrue,则对每个子层的输入进行层标准化(Layer Normalization),对每个子层的输出进行 dropout 和残差连接(residual connection)。否则(即 normalize_beforeFalse ),则对每个子层的输入不进行处理,只对每个子层的输出进行 dropout、残差连接(residual connection)和层标准化(Layer Normalization)。

参数

  • d_model (int) - 输入输出的维度。

  • nhead (int) - multi-head attention(MHA)的 Head 数量。

  • dim_feedforward (int) - 前馈神经网络中隐藏层的大小。

  • dropout_rate (float,可选) - 对两个子层的输出进行处理的 dropout 值,置零的概率。默认值:0.1。

  • activation (str,可选) - 前馈神经网络的激活函数。默认值:relu

  • attn_dropout_rate (float,可选) - MHA 中对注意力目标的随机失活率。如果为 Noneattn_dropout = dropout。默认值:None

  • act_dropout_rate (float,可选) - 前馈神经网络的激活函数后的 dropout 置零的概率。如果为 Noneact_dropout_rate = dropout_rate。默认值:None

  • normalize_before (bool,可选) - 设置对每个子层的输入输出的处理。如果为 True,则对每个子层的输入进行层标准化(Layer Normalization),否则(即为 False ),则对每个子层的输入不进行处理,而是在子层的输出前进行标准化。默认值:False

  • weight_attr (ParamAttr|tuple,可选) - 指定权重参数属性的对象。如果是 tuple ,MHA 的权重参数属性使用 weight_attr[0],前馈神经网络的权重参数属性使用 weight_attr[1]。如果参数值是 ParamAttr,则 MHA 和前馈神经网络的权重参数属性都使用 ParamAttr。默认值:None,表示使用默认的权重参数属性。具体用法请参见 ParamAttr

  • bias_attr (ParamAttr|tuple|bool,可选)- 指定偏置参数属性的对象。如果是 tuple ,MHA 的偏置参数属性使用 bias_attr[0],前馈神经网络的偏置参数属性使用 bias_attr[1]。如果该参数值是 ParamAttr,则 MHA 和前馈神经网络的偏置参数属性都使用 ParamAttr。如果该参数为 bool 类型,只支持为 False,表示没有偏置参数。默认值为 None,表示使用默认的偏置参数属性。具体用法请参见 ParamAttr

返回

  • Tensor,输出 Tensor,数据类型与 x 一样。

代码示例

>>> import paddle
>>> from paddle.incubate.nn import FusedTransformerEncoderLayer
>>> paddle.device.set_device('gpu')

>>> # encoder input: [batch_size, src_len, d_model]
>>> enc_input = paddle.rand((2, 4, 128))
>>> # self attention mask: [batch_size, n_head, src_len, src_len]
>>> attn_mask = paddle.rand((2, 2, 4, 4))
>>> encoder_layer = FusedTransformerEncoderLayer(128, 2, 512)
>>> enc_output = encoder_layer(enc_input, attn_mask)
>>> print(enc_output.shape)
[2, 4, 128]