forked from huawei/mindspore2022
!31075 fix mindspore.nn.transformer api name
Merge pull request !31075 from lvmingfu/master
This commit is contained in:
commit
e8889a81e7
|
|
@ -4,22 +4,22 @@ mindspore.nn.transformer
|
|||
.. note::
|
||||
Transformer网络。这些是实验性接口,可能会修改或删除。
|
||||
|
||||
.. include:: transformer/mindspore.nn.AttentionMask.rst
|
||||
.. include:: transformer/mindspore.nn.VocabEmbedding.rst
|
||||
.. include:: transformer/mindspore.nn.MultiHeadAttention.rst
|
||||
.. include:: transformer/mindspore.nn.FeedForward.rst
|
||||
.. include:: transformer/mindspore.nn.TransformerEncoder.rst
|
||||
.. include:: transformer/mindspore.nn.TransformerDecoder.rst
|
||||
.. include:: transformer/mindspore.nn.TransformerEncoderLayer.rst
|
||||
.. include:: transformer/mindspore.nn.TransformerDecoderLayer.rst
|
||||
.. include:: transformer/mindspore.nn.Transformer.rst
|
||||
.. include:: transformer/mindspore.nn.TransformerOpParallelConfig.rst
|
||||
.. include:: transformer/mindspore.nn.EmbeddingOpParallelConfig.rst
|
||||
.. include:: transformer/mindspore.nn.TransformerRecomputeConfig.rst
|
||||
.. include:: transformer/mindspore.nn.CrossEntropyLoss.rst
|
||||
.. include:: transformer/mindspore.nn.OpParallelConfig.rst
|
||||
.. include:: transformer/mindspore.nn.FixedSparseAttention.rst
|
||||
.. include:: transformer/mindspore.nn.MoEConfig.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.AttentionMask.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.VocabEmbedding.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.MultiHeadAttention.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.FeedForward.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.TransformerEncoder.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.TransformerDecoder.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.TransformerEncoderLayer.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.TransformerDecoderLayer.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.Transformer.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.TransformerOpParallelConfig.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.EmbeddingOpParallelConfig.rst
|
||||
.. include:: transformer/mindspore.nn.Ttransformer.ransformerRecomputeConfig.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.CrossEntropyLoss.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.OpParallelConfig.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.FixedSparseAttention.rst
|
||||
.. include:: transformer/mindspore.nn.transformer.MoEConfig.rst
|
||||
|
||||
.. automodule:: mindspore.nn.transformer
|
||||
:members:
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
.. py:class:: mindspore.nn.AttentionMask(seq_length, parallel_config=default_dpmp_config)
|
||||
.. py:class:: mindspore.nn.transformer.AttentionMask(seq_length, parallel_config=default_dpmp_config)
|
||||
|
||||
从输入掩码中获取下三角矩阵。输入掩码是值为1或0的二维Tensor (batch_size, seq_length)。1表示当前位置是一个有效的标记,其他值则表示当前位置不是一个有效的标记。
|
||||
|
||||
Loading…
Reference in New Issue