sequence_mask

paddle.fluid.layers.sequence_mask(x, maxlen=None, dtype=’int64’, name=None)[源代码]

该层根据输入 x 和 maxlen 输出一个掩码，数据类型为 dtype 。

假设 x 是一个形状为 [d_1, d_2，…, d_n] 的张量， 则输出 y 是一个形状为 [d_1, d_2，… ，d_n, maxlen] 的掩码，其中:

sequence_mask - 图1

范例如下：

  1. 给定输入：
  2.   x = [3, 1, 1, 0]  maxlen = 4
  4. 得到输出张量：
  5.   mask = [[1, 1, 1, 0],
  6.           [1, 0, 0, 0],
  7.           [1, 0, 0, 0],
  8.           [0, 0, 0, 0]]

参数

  • x (Variable) - 输入张量，其元素是小于等于 maxlen 的整数，形状为 [d_1, d_2，…, d_n] 的Tensor或LoDTensor。
  • maxlen (int，可选) - 序列的最大长度。默认为空，此时 maxlen 取 x 中所有元素的最大值。
  • dtype (np.dtype|core.VarDesc.VarType|str，可选) - 输出的数据类型，默认为 int64 。
  • name (str，可选) - 具体用法请参见 Name ，一般无需设置，默认值为None。

返回

mask张量，Tensor或LoDTensor，形状为 [d_1, d_2，… ，d_n, maxlen] ，数据类型由 dtype 指定，支持float32、float64、int32和int64，默认为int64。

返回类型

Variable

代码示例

  1. import paddle.fluid as fluid
  2. import paddle.fluid.layers as layers
  4. x = fluid.data(name='x', shape=[10], dtype='float32', lod_level=1)
  5. mask = layers.sequence_mask(x=x)