forked from huawei/mindspore2022
25 lines
1.1 KiB
ReStructuredText
25 lines
1.1 KiB
ReStructuredText
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.CrossEntropyLoss.rst
|
||
.. include:: transformer/mindspore.nn.OpParallelConfig.rst
|
||
.. include:: transformer/mindspore.nn.FixedSparseAttention.rst
|
||
.. include:: transformer/mindspore.nn.MoEConfig.rst
|
||
|
||
.. automodule:: mindspore.nn.transformer
|
||
:members:
|