paddlespeech.t2s.modules.transformer.mask 模块

掩码模块。

paddlespeech.t2s.modules.transformer.mask.subsequent_mask(size, dtype=paddle.bool)

为后续步骤创建掩码 (size, size)。

Args:

size (int):

掩模的大小

dtype (paddle.dtype):

结果 数据类型

Return:

Tensor:

>>> subsequent_mask(3)
[[1, 0, 0],
[1, 1, 0],
[1, 1, 1]]

paddlespeech.t2s.modules.transformer.mask.target_mask(ys_in_pad, ignore_id, dtype=paddle.bool)

为解码器自注意力创建掩码。

Args:

ys_pad (Tensor):

填充目标序列的批次 (B, Lmax)

ignore_id (int):

填充的索引

dtype (paddle.dtype):

结果 数据类型

Return:

张量: (B, Lmax, Lmax)