697 lines
17 KiB
ReStructuredText
697 lines
17 KiB
ReStructuredText
mindspore.mint
|
||
===============
|
||
|
||
mindspore.mint提供了大量的functional、nn、优化器接口，API用法及功能等与业界主流用法一致，方便用户参考使用。
|
||
mint接口当前是实验性接口，在图编译模式为O0和PyNative模式下性能比ops更优。当前暂不支持Ascend GE后端及CPU、GPU后端，后续会逐步完善。
|
||
|
||
模块导入方法如下：
|
||
|
||
.. code-block::
|
||
|
||
from mindspore import mint
|
||
|
||
MindSpore中 `mindspore.mint` 接口与上一版本相比，新增、删除和支持平台的变化信息请参考 `mindspore.mint API接口变更 <https://gitee.com/mindspore/docs/blob/master/resource/api_updates/mint_api_updates_cn.md>`_ 。
|
||
|
||
Tensor
|
||
---------------
|
||
|
||
创建运算
|
||
^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.arange
|
||
mindspore.mint.empty
|
||
mindspore.mint.empty_like
|
||
mindspore.mint.eye
|
||
mindspore.mint.full
|
||
mindspore.mint.full_like
|
||
mindspore.mint.linspace
|
||
mindspore.mint.ones
|
||
mindspore.mint.ones_like
|
||
mindspore.mint.polar
|
||
mindspore.mint.zeros
|
||
mindspore.mint.zeros_like
|
||
|
||
索引、切分、连接、突变运算
|
||
^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.cat
|
||
mindspore.mint.chunk
|
||
mindspore.mint.concat
|
||
mindspore.mint.gather
|
||
mindspore.mint.index_add
|
||
mindspore.mint.index_select
|
||
mindspore.mint.masked_select
|
||
mindspore.mint.narrow
|
||
mindspore.mint.nonzero
|
||
mindspore.mint.permute
|
||
mindspore.mint.reshape
|
||
mindspore.mint.scatter
|
||
mindspore.mint.scatter_add
|
||
mindspore.mint.select
|
||
mindspore.mint.split
|
||
mindspore.mint.squeeze
|
||
mindspore.mint.stack
|
||
mindspore.mint.swapaxes
|
||
mindspore.mint.tile
|
||
mindspore.mint.transpose
|
||
mindspore.mint.unbind
|
||
mindspore.mint.unsqueeze
|
||
mindspore.mint.where
|
||
|
||
随机采样
|
||
------------
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.bernoulli
|
||
mindspore.mint.multinomial
|
||
mindspore.mint.normal
|
||
mindspore.mint.rand
|
||
mindspore.mint.rand_like
|
||
mindspore.mint.randint
|
||
mindspore.mint.randint_like
|
||
mindspore.mint.randn
|
||
mindspore.mint.randn_like
|
||
mindspore.mint.randperm
|
||
|
||
数学运算
|
||
------------------
|
||
|
||
逐元素运算
|
||
^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.abs
|
||
mindspore.mint.acos
|
||
mindspore.mint.acosh
|
||
mindspore.mint.add
|
||
mindspore.mint.arccos
|
||
mindspore.mint.arccosh
|
||
mindspore.mint.arcsin
|
||
mindspore.mint.arcsinh
|
||
mindspore.mint.arctan
|
||
mindspore.mint.arctan2
|
||
mindspore.mint.arctanh
|
||
mindspore.mint.asin
|
||
mindspore.mint.asinh
|
||
mindspore.mint.atan
|
||
mindspore.mint.atan2
|
||
mindspore.mint.atanh
|
||
mindspore.mint.bitwise_and
|
||
mindspore.mint.bitwise_or
|
||
mindspore.mint.bitwise_xor
|
||
mindspore.mint.ceil
|
||
mindspore.mint.clamp
|
||
mindspore.mint.cos
|
||
mindspore.mint.cosh
|
||
mindspore.mint.div
|
||
mindspore.mint.divide
|
||
mindspore.mint.erf
|
||
mindspore.mint.erfc
|
||
mindspore.mint.erfinv
|
||
mindspore.mint.exp
|
||
mindspore.mint.exp2
|
||
mindspore.mint.expm1
|
||
mindspore.mint.fix
|
||
mindspore.mint.float_power
|
||
mindspore.mint.floor
|
||
mindspore.mint.floor_divide
|
||
mindspore.mint.fmod
|
||
mindspore.mint.frac
|
||
mindspore.mint.imag
|
||
mindspore.mint.lerp
|
||
mindspore.mint.log
|
||
mindspore.mint.log10
|
||
mindspore.mint.log1p
|
||
mindspore.mint.log2
|
||
mindspore.mint.logaddexp
|
||
mindspore.mint.logaddexp2
|
||
mindspore.mint.logical_and
|
||
mindspore.mint.logical_not
|
||
mindspore.mint.logical_or
|
||
mindspore.mint.logical_xor
|
||
mindspore.mint.mul
|
||
mindspore.mint.nan_to_num
|
||
mindspore.mint.neg
|
||
mindspore.mint.negative
|
||
mindspore.mint.pow
|
||
mindspore.mint.real
|
||
mindspore.mint.reciprocal
|
||
mindspore.mint.remainder
|
||
mindspore.mint.round
|
||
mindspore.mint.rsqrt
|
||
mindspore.mint.sigmoid
|
||
mindspore.mint.sign
|
||
mindspore.mint.sin
|
||
mindspore.mint.sinc
|
||
mindspore.mint.sinh
|
||
mindspore.mint.softmax
|
||
mindspore.mint.sqrt
|
||
mindspore.mint.square
|
||
mindspore.mint.sub
|
||
mindspore.mint.t
|
||
mindspore.mint.tan
|
||
mindspore.mint.tanh
|
||
mindspore.mint.trunc
|
||
mindspore.mint.xlogy
|
||
|
||
Reduction运算
|
||
^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.all
|
||
mindspore.mint.amax
|
||
mindspore.mint.amin
|
||
mindspore.mint.any
|
||
mindspore.mint.argmax
|
||
mindspore.mint.argmin
|
||
mindspore.mint.count_nonzero
|
||
mindspore.mint.logsumexp
|
||
mindspore.mint.max
|
||
mindspore.mint.mean
|
||
mindspore.mint.median
|
||
mindspore.mint.min
|
||
mindspore.mint.nansum
|
||
mindspore.mint.norm
|
||
mindspore.mint.prod
|
||
mindspore.mint.std
|
||
mindspore.mint.std_mean
|
||
mindspore.mint.sum
|
||
mindspore.mint.unique
|
||
mindspore.mint.unique_consecutive
|
||
mindspore.mint.var
|
||
mindspore.mint.var_mean
|
||
|
||
比较运算
|
||
^^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.allclose
|
||
mindspore.mint.argsort
|
||
mindspore.mint.eq
|
||
mindspore.mint.equal
|
||
mindspore.mint.greater
|
||
mindspore.mint.greater_equal
|
||
mindspore.mint.gt
|
||
mindspore.mint.isclose
|
||
mindspore.mint.isfinite
|
||
mindspore.mint.isinf
|
||
mindspore.mint.isneginf
|
||
mindspore.mint.le
|
||
mindspore.mint.less
|
||
mindspore.mint.less_equal
|
||
mindspore.mint.lt
|
||
mindspore.mint.maximum
|
||
mindspore.mint.minimum
|
||
mindspore.mint.ne
|
||
mindspore.mint.not_equal
|
||
mindspore.mint.sort
|
||
mindspore.mint.topk
|
||
|
||
BLAS和LAPACK运算
|
||
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.addbmm
|
||
mindspore.mint.addmm
|
||
mindspore.mint.addmv
|
||
mindspore.mint.baddbmm
|
||
mindspore.mint.bmm
|
||
mindspore.mint.dot
|
||
mindspore.mint.inverse
|
||
mindspore.mint.matmul
|
||
mindspore.mint.mm
|
||
mindspore.mint.mv
|
||
mindspore.mint.outer
|
||
mindspore.mint.triangular_solve
|
||
|
||
其他运算
|
||
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.bincount
|
||
mindspore.mint.broadcast_to
|
||
mindspore.mint.cdist
|
||
mindspore.mint.clone
|
||
mindspore.mint.cross
|
||
mindspore.mint.cummax
|
||
mindspore.mint.cummin
|
||
mindspore.mint.cumprod
|
||
mindspore.mint.cumsum
|
||
mindspore.mint.diag
|
||
mindspore.mint.diff
|
||
mindspore.mint.einsum
|
||
mindspore.mint.flatten
|
||
mindspore.mint.flip
|
||
mindspore.mint.histc
|
||
mindspore.mint.meshgrid
|
||
mindspore.mint.ravel
|
||
mindspore.mint.repeat_interleave
|
||
mindspore.mint.roll
|
||
mindspore.mint.searchsorted
|
||
mindspore.mint.trace
|
||
mindspore.mint.tril
|
||
mindspore.mint.triu
|
||
|
||
mindspore.mint.nn
|
||
------------------
|
||
|
||
卷积层
|
||
^^^^^^^^^^^^^^^^^^
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.Conv1d
|
||
mindspore.mint.nn.Conv2d
|
||
mindspore.mint.nn.Conv3d
|
||
mindspore.mint.nn.ConvTranspose2d
|
||
mindspore.mint.nn.Fold
|
||
mindspore.mint.nn.Unfold
|
||
|
||
池化层
|
||
^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.AdaptiveAvgPool1d
|
||
mindspore.mint.nn.AdaptiveAvgPool2d
|
||
mindspore.mint.nn.AdaptiveAvgPool3d
|
||
mindspore.mint.nn.AdaptiveMaxPool1d
|
||
mindspore.mint.nn.AdaptiveMaxPool2d
|
||
mindspore.mint.nn.AvgPool2d
|
||
mindspore.mint.nn.AvgPool3d
|
||
mindspore.mint.nn.MaxUnpool2d
|
||
|
||
填充层
|
||
^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.ConstantPad1d
|
||
mindspore.mint.nn.ConstantPad2d
|
||
mindspore.mint.nn.ConstantPad3d
|
||
mindspore.mint.nn.ReflectionPad1d
|
||
mindspore.mint.nn.ReflectionPad2d
|
||
mindspore.mint.nn.ReflectionPad3d
|
||
mindspore.mint.nn.ReplicationPad1d
|
||
mindspore.mint.nn.ReplicationPad2d
|
||
mindspore.mint.nn.ReplicationPad3d
|
||
mindspore.mint.nn.ZeroPad1d
|
||
mindspore.mint.nn.ZeroPad2d
|
||
mindspore.mint.nn.ZeroPad3d
|
||
|
||
非线性激活层 (加权和，非线性)
|
||
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.ELU
|
||
mindspore.mint.nn.GELU
|
||
mindspore.mint.nn.GLU
|
||
mindspore.mint.nn.Hardshrink
|
||
mindspore.mint.nn.Hardsigmoid
|
||
mindspore.mint.nn.Hardswish
|
||
mindspore.mint.nn.LogSigmoid
|
||
mindspore.mint.nn.LogSoftmax
|
||
mindspore.mint.nn.Mish
|
||
mindspore.mint.nn.PReLU
|
||
mindspore.mint.nn.ReLU
|
||
mindspore.mint.nn.ReLU6
|
||
mindspore.mint.nn.SELU
|
||
mindspore.mint.nn.Sigmoid
|
||
mindspore.mint.nn.SiLU
|
||
mindspore.mint.nn.Softmax
|
||
mindspore.mint.nn.Softshrink
|
||
mindspore.mint.nn.Tanh
|
||
mindspore.mint.nn.Threshold
|
||
|
||
归一化层
|
||
^^^^^^^^^^^^^^^^^^
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.BatchNorm1d
|
||
mindspore.mint.nn.BatchNorm2d
|
||
mindspore.mint.nn.BatchNorm3d
|
||
mindspore.mint.nn.GroupNorm
|
||
mindspore.mint.nn.LayerNorm
|
||
mindspore.mint.nn.SyncBatchNorm
|
||
|
||
线性层
|
||
^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.Identity
|
||
mindspore.mint.nn.Linear
|
||
|
||
Dropout层
|
||
^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.Dropout
|
||
mindspore.mint.nn.Dropout2d
|
||
|
||
稀疏层
|
||
^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.Embedding
|
||
|
||
损失函数
|
||
^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.BCELoss
|
||
mindspore.mint.nn.BCEWithLogitsLoss
|
||
mindspore.mint.nn.CosineEmbeddingLoss
|
||
mindspore.mint.nn.CrossEntropyLoss
|
||
mindspore.mint.nn.KLDivLoss
|
||
mindspore.mint.nn.L1Loss
|
||
mindspore.mint.nn.MSELoss
|
||
mindspore.mint.nn.NLLLoss
|
||
mindspore.mint.nn.SmoothL1Loss
|
||
|
||
Vision层
|
||
^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.PixelShuffle
|
||
mindspore.mint.nn.Upsample
|
||
|
||
mindspore.mint.nn.functional
|
||
-----------------------------
|
||
|
||
卷积函数
|
||
^^^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.functional.conv1d
|
||
mindspore.mint.nn.functional.conv2d
|
||
mindspore.mint.nn.functional.conv3d
|
||
mindspore.mint.nn.functional.conv_transpose2d
|
||
mindspore.mint.nn.functional.fold
|
||
mindspore.mint.nn.functional.unfold
|
||
|
||
池化函数
|
||
^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.functional.adaptive_avg_pool1d
|
||
mindspore.mint.nn.functional.adaptive_avg_pool2d
|
||
mindspore.mint.nn.functional.adaptive_avg_pool3d
|
||
mindspore.mint.nn.functional.adaptive_max_pool1d
|
||
mindspore.mint.nn.functional.adaptive_max_pool2d
|
||
mindspore.mint.nn.functional.avg_pool1d
|
||
mindspore.mint.nn.functional.avg_pool2d
|
||
mindspore.mint.nn.functional.avg_pool3d
|
||
mindspore.mint.nn.functional.max_pool2d
|
||
mindspore.mint.nn.functional.max_unpool2d
|
||
|
||
非线性激活函数
|
||
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.functional.batch_norm
|
||
mindspore.mint.nn.functional.elu
|
||
mindspore.mint.nn.functional.elu_
|
||
mindspore.mint.nn.functional.gelu
|
||
mindspore.mint.nn.functional.glu
|
||
mindspore.mint.nn.functional.group_norm
|
||
mindspore.mint.nn.functional.hardshrink
|
||
mindspore.mint.nn.functional.hardsigmoid
|
||
mindspore.mint.nn.functional.hardswish
|
||
mindspore.mint.nn.functional.layer_norm
|
||
mindspore.mint.nn.functional.leaky_relu
|
||
mindspore.mint.nn.functional.log_softmax
|
||
mindspore.mint.nn.functional.logsigmoid
|
||
mindspore.mint.nn.functional.mish
|
||
mindspore.mint.nn.functional.normalize
|
||
mindspore.mint.nn.functional.prelu
|
||
mindspore.mint.nn.functional.relu
|
||
mindspore.mint.nn.functional.relu_
|
||
mindspore.mint.nn.functional.relu6
|
||
mindspore.mint.nn.functional.selu
|
||
mindspore.mint.nn.functional.sigmoid
|
||
mindspore.mint.nn.functional.silu
|
||
mindspore.mint.nn.functional.softmax
|
||
mindspore.mint.nn.functional.softplus
|
||
mindspore.mint.nn.functional.softshrink
|
||
mindspore.mint.nn.functional.tanh
|
||
mindspore.mint.nn.functional.threshold
|
||
mindspore.mint.nn.functional.threshold_
|
||
|
||
线性函数
|
||
^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.functional.linear
|
||
|
||
Dropout函数
|
||
^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.functional.dropout
|
||
mindspore.mint.nn.functional.dropout2d
|
||
|
||
稀疏函数
|
||
^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.functional.embedding
|
||
mindspore.mint.nn.functional.one_hot
|
||
|
||
损失函数
|
||
^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.functional.binary_cross_entropy
|
||
mindspore.mint.nn.functional.binary_cross_entropy_with_logits
|
||
mindspore.mint.nn.functional.cosine_embedding_loss
|
||
mindspore.mint.nn.functional.cross_entropy
|
||
mindspore.mint.nn.functional.kl_div
|
||
mindspore.mint.nn.functional.l1_loss
|
||
mindspore.mint.nn.functional.mse_loss
|
||
mindspore.mint.nn.functional.nll_loss
|
||
mindspore.mint.nn.functional.smooth_l1_loss
|
||
|
||
Vision函数
|
||
^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.nn.functional.grid_sample
|
||
mindspore.mint.nn.functional.interpolate
|
||
mindspore.mint.nn.functional.pad
|
||
mindspore.mint.nn.functional.pixel_shuffle
|
||
|
||
mindspore.mint.optim
|
||
---------------------
|
||
|
||
算法
|
||
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.optim.Adam
|
||
mindspore.mint.optim.AdamW
|
||
mindspore.mint.optim.SGD
|
||
|
||
mindspore.mint.linalg
|
||
----------------------
|
||
|
||
矩阵属性
|
||
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.linalg.matrix_norm
|
||
mindspore.mint.linalg.norm
|
||
mindspore.mint.linalg.vector_norm
|
||
|
||
分解
|
||
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.linalg.qr
|
||
|
||
逆数
|
||
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.linalg.inv
|
||
|
||
mindspore.mint.special
|
||
----------------------
|
||
|
||
逐元素运算
|
||
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.special.erfc
|
||
mindspore.mint.special.exp2
|
||
mindspore.mint.special.expm1
|
||
mindspore.mint.special.log1p
|
||
mindspore.mint.special.log_softmax
|
||
mindspore.mint.special.round
|
||
mindspore.mint.special.sinc
|
||
|
||
mindspore.mint.distributed
|
||
--------------------------------
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: mint
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.mint.distributed.all_gather
|
||
mindspore.mint.distributed.all_gather_into_tensor
|
||
mindspore.mint.distributed.all_gather_into_tensor_uneven
|
||
mindspore.mint.distributed.all_gather_object
|
||
mindspore.mint.distributed.all_reduce
|
||
mindspore.mint.distributed.all_to_all
|
||
mindspore.mint.distributed.all_to_all_single
|
||
mindspore.mint.distributed.barrier
|
||
mindspore.mint.distributed.batch_isend_irecv
|
||
mindspore.mint.distributed.broadcast
|
||
mindspore.mint.distributed.broadcast_object_list
|
||
mindspore.mint.distributed.destroy_process_group
|
||
mindspore.mint.distributed.gather
|
||
mindspore.mint.distributed.gather_object
|
||
mindspore.mint.distributed.get_backend
|
||
mindspore.mint.distributed.get_global_rank
|
||
mindspore.mint.distributed.get_group_rank
|
||
mindspore.mint.distributed.get_process_group_ranks
|
||
mindspore.mint.distributed.get_rank
|
||
mindspore.mint.distributed.get_world_size
|
||
mindspore.mint.distributed.init_process_group
|
||
mindspore.mint.distributed.irecv
|
||
mindspore.mint.distributed.isend
|
||
mindspore.mint.distributed.is_available
|
||
mindspore.mint.distributed.is_initialized
|
||
mindspore.mint.distributed.new_group
|
||
mindspore.mint.distributed.P2POp
|
||
mindspore.mint.distributed.recv
|
||
mindspore.mint.distributed.reduce
|
||
mindspore.mint.distributed.reduce_scatter
|
||
mindspore.mint.distributed.reduce_scatter_tensor
|
||
mindspore.mint.distributed.reduce_scatter_tensor_uneven
|
||
mindspore.mint.distributed.scatter
|
||
mindspore.mint.distributed.scatter_object_list
|
||
mindspore.mint.distributed.send
|
||
mindspore.mint.distributed.TCPStore
|