forked from huawei/mindspore2022
23 lines
1.0 KiB
ReStructuredText
23 lines
1.0 KiB
ReStructuredText
.. py:class:: mindspore.nn.AttentionMask(seq_length, parallel_config=default_dpmp_config)
|
||
|
||
从输入掩码中获取下三角矩阵。输入掩码是值为1或0的二维Tensor (batch_size, seq_length)。1表示当前位置是一个有效的标记，其他值则表示当前位置不是一个有效的标记。
|
||
|
||
**参数：**
|
||
|
||
- **seq_length** (int) - 表示输入Tensor的序列长度。
|
||
- **parallel_config** (OpParallelConfig) - 表示并行配置。默认值为 `default_dpmp_config` ，表示一个带有默认参数的 `OpParallelConfig` 实例。
|
||
|
||
**输入：**
|
||
|
||
- **input_mask** (Tensor) - 掩码矩阵，shape为(batch_size, seq_length)，表示每个位置是否为有效输入。
|
||
|
||
**输出：**
|
||
|
||
Tensor，表示shape为(batch_size, seq_length, seq_length)的注意力掩码矩阵。
|
||
|
||
**异常：**
|
||
|
||
- **TypeError** - `seq_length` 不是整数。
|
||
- **ValueError** - `seq_length` 不是正数。
|
||
- **TypeError** - `parallel_config` 不是OpParallelConfig的子类。
|