zzy34407230
/
mindspore2022
Not watched
1
0
Fork 0
Code
Releases 22 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: b67f19b9a0
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'b67f19b9a0'
${ noResults }
mindspore2022/docs/api/api_python/operations.rst

433 lines
12 kB
Raw Blame History 

  1. operations

  2. ----------

  3. 

  4. 神经网络算子

  5. ^^^^^^^^^^^^^^^^^^^^^^^^

  6. 

  7. operations中的Primitive算子在使用前需要实例化。

  8. 

  9. .. mscnplatformautosummary::

  10.     :toctree: ops

  11.     :nosignatures:

  12.     :template: classtemplate.rst

  13. 

  14.     mindspore.ops.Acosh

  15.     mindspore.ops.Adam

  16.     mindspore.ops.AdamNoUpdateParam

  17.     mindspore.ops.AdamWeightDecay

  18.     mindspore.ops.AdaptiveAvgPool2D

  19.     mindspore.ops.ApplyAdadelta

  20.     mindspore.ops.ApplyAdagrad

  21.     mindspore.ops.ApplyAdagradDA

  22.     mindspore.ops.ApplyAdagradV2

  23.     mindspore.ops.ApplyAdaMax

  24.     mindspore.ops.ApplyAddSign

  25.     mindspore.ops.ApplyCenteredRMSProp

  26.     mindspore.ops.ApplyGradientDescent

  27.     mindspore.ops.ApplyMomentum

  28.     mindspore.ops.ApplyPowerSign

  29.     mindspore.ops.ApplyProximalAdagrad

  30.     mindspore.ops.ApplyProximalGradientDescent

  31.     mindspore.ops.ApplyRMSProp

  32.     mindspore.ops.AvgPool

  33.     mindspore.ops.AvgPool3D

  34.     mindspore.ops.BasicLSTMCell

  35.     mindspore.ops.BatchNorm

  36.     mindspore.ops.BCEWithLogitsLoss

  37.     mindspore.ops.BiasAdd

  38.     mindspore.ops.BinaryCrossEntropy

  39.     mindspore.ops.ComputeAccidentalHits

  40.     mindspore.ops.Conv2D

  41.     mindspore.ops.Conv2DBackpropInput

  42.     mindspore.ops.Conv2DTranspose

  43.     mindspore.ops.Conv3D

  44.     mindspore.ops.Conv3DTranspose

  45.     mindspore.ops.CTCGreedyDecoder

  46.     mindspore.ops.CTCLoss

  47.     mindspore.ops.DataFormatDimMap

  48.     mindspore.ops.DepthwiseConv2dNative

  49.     mindspore.ops.Dropout

  50.     mindspore.ops.Dropout2D

  51.     mindspore.ops.Dropout3D

  52.     mindspore.ops.DropoutDoMask

  53.     mindspore.ops.DropoutGenMask

  54.     mindspore.ops.DynamicGRUV2

  55.     mindspore.ops.DynamicRNN

  56.     mindspore.ops.Elu

  57.     mindspore.ops.FastGeLU

  58.     mindspore.ops.Flatten

  59.     mindspore.ops.FloorMod

  60.     mindspore.ops.FusedSparseAdam

  61.     mindspore.ops.FusedSparseLazyAdam

  62.     mindspore.ops.FusedSparseProximalAdagrad

  63.     mindspore.ops.GeLU

  64.     mindspore.ops.GetNext

  65.     mindspore.ops.HShrink

  66.     mindspore.ops.HSigmoid

  67.     mindspore.ops.HSwish

  68.     mindspore.ops.KLDivLoss

  69.     mindspore.ops.L2Loss

  70.     mindspore.ops.L2Normalize

  71.     mindspore.ops.LARSUpdate

  72.     mindspore.ops.LayerNorm

  73.     mindspore.ops.LogSoftmax

  74.     mindspore.ops.LRN

  75.     mindspore.ops.LSTM

  76.     mindspore.ops.MaxPool

  77.     mindspore.ops.MaxPool3D

  78.     mindspore.ops.MaxPoolWithArgmax

  79.     mindspore.ops.MirrorPad

  80.     mindspore.ops.Mish

  81.     mindspore.ops.NLLLoss

  82.     mindspore.ops.OneHot

  83.     mindspore.ops.Pad

  84.     mindspore.ops.PReLU

  85.     mindspore.ops.ReLU

  86.     mindspore.ops.ReLU6

  87.     mindspore.ops.ReLUV2

  88.     mindspore.ops.ResizeBilinear

  89.     mindspore.ops.RNNTLoss

  90.     mindspore.ops.ROIAlign

  91.     mindspore.ops.SeLU

  92.     mindspore.ops.SGD

  93.     mindspore.ops.Sigmoid

  94.     mindspore.ops.SigmoidCrossEntropyWithLogits

  95.     mindspore.ops.SmoothL1Loss

  96.     mindspore.ops.SoftMarginLoss

  97.     mindspore.ops.Softmax

  98.     mindspore.ops.SoftmaxCrossEntropyWithLogits

  99.     mindspore.ops.Softplus

  100.     mindspore.ops.SoftShrink

  101.     mindspore.ops.Softsign

  102.     mindspore.ops.SparseApplyAdagrad

  103.     mindspore.ops.SparseApplyAdagradV2

  104.     mindspore.ops.SparseApplyProximalAdagrad

  105.     mindspore.ops.SparseSoftmaxCrossEntropyWithLogits

  106.     mindspore.ops.Stack

  107.     mindspore.ops.Tanh

  108.     mindspore.ops.TopK

  109.     mindspore.ops.Unstack

  110. 

  111. 数学算子

  112. ^^^^^^^^^^^^^^^

  113. 

  114. .. mscnplatformautosummary::

  115.     :toctree: ops

  116.     :nosignatures:

  117.     :template: classtemplate.rst

  118. 

  119.     mindspore.ops.Abs

  120.     mindspore.ops.AccumulateNV2

  121.     mindspore.ops.ACos

  122.     mindspore.ops.Add

  123.     mindspore.ops.AddN

  124.     mindspore.ops.ApproximateEqual

  125.     mindspore.ops.Asin

  126.     mindspore.ops.Asinh

  127.     mindspore.ops.AssignAdd

  128.     mindspore.ops.AssignSub

  129.     mindspore.ops.Atan

  130.     mindspore.ops.Atan2

  131.     mindspore.ops.Atanh

  132.     mindspore.ops.BatchMatMul

  133.     mindspore.ops.BesselI0e

  134.     mindspore.ops.BesselI1e

  135.     mindspore.ops.BitwiseAnd

  136.     mindspore.ops.BitwiseOr

  137.     mindspore.ops.BitwiseXor

  138.     mindspore.ops.Cdist

  139.     mindspore.ops.Ceil

  140.     mindspore.ops.Conj

  141.     mindspore.ops.Cos

  142.     mindspore.ops.Cosh

  143.     mindspore.ops.CumProd

  144.     mindspore.ops.CumSum

  145.     mindspore.ops.Div

  146.     mindspore.ops.DivNoNan

  147.     mindspore.ops.Eps

  148.     mindspore.ops.Equal

  149.     mindspore.ops.EqualCount

  150.     mindspore.ops.Erf

  151.     mindspore.ops.Erfc

  152.     mindspore.ops.Erfinv

  153.     mindspore.ops.Exp

  154.     mindspore.ops.Expm1

  155.     mindspore.ops.FloatStatus

  156.     mindspore.ops.Floor

  157.     mindspore.ops.FloorDiv

  158.     mindspore.ops.Ger

  159.     mindspore.ops.Greater

  160.     mindspore.ops.GreaterEqual

  161.     mindspore.ops.HistogramFixedWidth

  162.     mindspore.ops.Imag

  163.     mindspore.ops.IndexAdd

  164.     mindspore.ops.InplaceAdd

  165.     mindspore.ops.InplaceSub

  166.     mindspore.ops.Inv

  167.     mindspore.ops.Invert

  168.     mindspore.ops.IsInf

  169.     mindspore.ops.IsNan

  170.     mindspore.ops.Lerp

  171.     mindspore.ops.Less

  172.     mindspore.ops.LessEqual

  173.     mindspore.ops.LinSpace

  174.     mindspore.ops.Log

  175.     mindspore.ops.Log1p

  176.     mindspore.ops.LogicalAnd

  177.     mindspore.ops.LogicalNot

  178.     mindspore.ops.LogicalOr

  179.     mindspore.ops.LpNorm

  180.     mindspore.ops.MatMul

  181.     mindspore.ops.MatrixInverse

  182.     mindspore.ops.Maximum

  183.     mindspore.ops.Minimum

  184.     mindspore.ops.Mod

  185.     mindspore.ops.Mul

  186.     mindspore.ops.MulNoNan

  187.     mindspore.ops.Neg

  188.     mindspore.ops.NMSWithMask

  189.     mindspore.ops.NotEqual

  190.     mindspore.ops.NPUAllocFloatStatus

  191.     mindspore.ops.NPUClearFloatStatus

  192.     mindspore.ops.NPUGetFloatStatus

  193.     mindspore.ops.Pow

  194.     mindspore.ops.Real

  195.     mindspore.ops.RealDiv

  196.     mindspore.ops.Reciprocal

  197.     mindspore.ops.ReduceAll

  198.     mindspore.ops.ReduceAny

  199.     mindspore.ops.ReduceMax

  200.     mindspore.ops.ReduceMean

  201.     mindspore.ops.ReduceMin

  202.     mindspore.ops.ReduceProd

  203.     mindspore.ops.ReduceSum

  204.     mindspore.ops.Round

  205.     mindspore.ops.Rsqrt

  206.     mindspore.ops.Sign

  207.     mindspore.ops.Sin

  208.     mindspore.ops.Sinh

  209.     mindspore.ops.Sqrt

  210.     mindspore.ops.Square

  211.     mindspore.ops.SquaredDifference

  212.     mindspore.ops.SquareSumAll

  213.     mindspore.ops.Sub

  214.     mindspore.ops.Tan

  215.     mindspore.ops.TruncateDiv

  216.     mindspore.ops.TruncateMod

  217.     mindspore.ops.Xdivy

  218.     mindspore.ops.Xlogy

  219. 

  220. 数组算子

  221. ^^^^^^^^^^^^^^^^

  222. 

  223. .. mscnplatformautosummary::

  224.     :toctree: ops

  225.     :nosignatures:

  226.     :template: classtemplate.rst

  227. 

  228.     mindspore.ops.ApplyFtrl

  229.     mindspore.ops.Argmax

  230.     mindspore.ops.ArgMaxWithValue

  231.     mindspore.ops.Argmin

  232.     mindspore.ops.ArgMinWithValue

  233.     mindspore.ops.BatchToSpace

  234.     mindspore.ops.BatchToSpaceND

  235.     mindspore.ops.BroadcastTo

  236.     mindspore.ops.Cast

  237.     mindspore.ops.Concat

  238.     mindspore.ops.DepthToSpace

  239.     mindspore.ops.DType

  240.     mindspore.ops.DynamicShape

  241.     mindspore.ops.EditDistance

  242.     mindspore.ops.EmbeddingLookup

  243.     mindspore.ops.ExpandDims

  244.     mindspore.ops.ExtractVolumePatches

  245.     mindspore.ops.Eye

  246.     mindspore.ops.Fill

  247.     mindspore.ops.FusedSparseFtrl

  248.     mindspore.ops.Gather

  249.     mindspore.ops.GatherD

  250.     mindspore.ops.GatherNd

  251.     mindspore.ops.Identity

  252.     mindspore.ops.InplaceUpdate

  253.     mindspore.ops.InvertPermutation

  254.     mindspore.ops.IsFinite

  255.     mindspore.ops.IsInstance

  256.     mindspore.ops.IsSubClass

  257.     mindspore.ops.MaskedFill

  258.     mindspore.ops.MaskedSelect

  259.     mindspore.ops.Meshgrid

  260.     mindspore.ops.Ones

  261.     mindspore.ops.OnesLike

  262.     mindspore.ops.Padding

  263.     mindspore.ops.ParallelConcat

  264.     mindspore.ops.Randperm

  265.     mindspore.ops.Rank

  266.     mindspore.ops.Reshape

  267.     mindspore.ops.ResizeNearestNeighbor

  268.     mindspore.ops.ReverseSequence

  269.     mindspore.ops.ReverseV2

  270.     mindspore.ops.Rint

  271.     mindspore.ops.SameTypeShape

  272.     mindspore.ops.ScalarCast

  273.     mindspore.ops.ScalarToArray

  274.     mindspore.ops.ScalarToTensor

  275.     mindspore.ops.ScatterAdd

  276.     mindspore.ops.ScatterDiv

  277.     mindspore.ops.ScatterMax

  278.     mindspore.ops.ScatterMin

  279.     mindspore.ops.ScatterMul

  280.     mindspore.ops.ScatterNd

  281.     mindspore.ops.ScatterNdAdd

  282.     mindspore.ops.ScatterNdSub

  283.     mindspore.ops.ScatterNdUpdate

  284.     mindspore.ops.ScatterNonAliasingAdd

  285.     mindspore.ops.ScatterSub

  286.     mindspore.ops.ScatterUpdate

  287.     mindspore.ops.Select

  288.     mindspore.ops.Shape

  289.     mindspore.ops.Size

  290.     mindspore.ops.Slice

  291.     mindspore.ops.Sort

  292.     mindspore.ops.SpaceToBatch

  293.     mindspore.ops.SpaceToBatchND

  294.     mindspore.ops.SpaceToDepth

  295.     mindspore.ops.SparseApplyFtrl

  296.     mindspore.ops.SparseApplyFtrlV2

  297.     mindspore.ops.SparseGatherV2

  298.     mindspore.ops.Split

  299.     mindspore.ops.SplitV

  300.     mindspore.ops.Squeeze

  301.     mindspore.ops.StridedSlice

  302.     mindspore.ops.TensorScatterAdd

  303.     mindspore.ops.TensorScatterMax

  304.     mindspore.ops.TensorScatterMin

  305.     mindspore.ops.TensorScatterSub

  306.     mindspore.ops.TensorScatterUpdate

  307.     mindspore.ops.TensorShape

  308.     mindspore.ops.Tile

  309.     mindspore.ops.Transpose

  310.     mindspore.ops.TupleToArray

  311.     mindspore.ops.Unique

  312.     mindspore.ops.UniqueWithPad

  313.     mindspore.ops.UnsortedSegmentMax

  314.     mindspore.ops.UnsortedSegmentMin

  315.     mindspore.ops.UnsortedSegmentProd

  316.     mindspore.ops.UnsortedSegmentSum

  317.     mindspore.ops.Zeros

  318.     mindspore.ops.ZerosLike

  319. 

  320. 通信算子

  321. ^^^^^^^^^^^^^^^^^^^^^^^

  322. 

  323. 注意，以下列表中的接口需要先配置好通信环境变量。

  324. 

  325. 针对Ascend设备，用户需要准备rank表，设置rank_id和device_id，详见 `Ascend指导文档 \

  326. <https://www.mindspore.cn/tutorials/zh-CN/master/intermediate/distributed_training/

  327. distributed_training_ascend.html#id12>`_ 。

  328. 

  329. 针对GPU设备，用户需要准备host文件和mpi，详见 `GPU指导文档 \

  330. <https://www.mindspore.cn/tutorials/zh-CN/master/intermediate/distributed_training/distributed_training_gpu.html#id6>`_ 。

  331. 

  332. .. mscnplatformautosummary::

  333.     :toctree: ops

  334.     :nosignatures:

  335.     :template: classtemplate.rst

  336. 

  337.     mindspore.ops.AllGather

  338.     mindspore.ops.AllReduce

  339.     mindspore.ops.AlltoAll

  340.     mindspore.ops.Broadcast

  341.     mindspore.ops.NeighborExchange

  342.     mindspore.ops.NeighborExchangeV2

  343.     mindspore.ops.ReduceOp

  344.     mindspore.ops.ReduceScatter

  345. 

  346. 调试算子

  347. ^^^^^^^^^^^^^^^

  348. 

  349. .. mscnplatformautosummary::

  350.     :toctree: ops

  351.     :nosignatures:

  352.     :template: classtemplate.rst

  353. 

  354.     mindspore.ops.HistogramSummary

  355.     mindspore.ops.HookBackward

  356.     mindspore.ops.ImageSummary

  357.     mindspore.ops.InsertGradientOf

  358.     mindspore.ops.Print

  359.     mindspore.ops.ScalarSummary

  360.     mindspore.ops.TensorSummary

  361. 

  362. 

  363. 随机算子

  364. ^^^^^^^^^^^^^^^^^

  365. 

  366. .. mscnplatformautosummary::

  367.     :toctree: ops

  368.     :nosignatures:

  369.     :template: classtemplate.rst

  370. 

  371.     mindspore.ops.Gamma

  372.     mindspore.ops.LogUniformCandidateSampler

  373.     mindspore.ops.Multinomial

  374.     mindspore.ops.Poisson

  375.     mindspore.ops.RandomCategorical

  376.     mindspore.ops.RandomChoiceWithMask

  377.     mindspore.ops.StandardLaplace

  378.     mindspore.ops.StandardNormal

  379.     mindspore.ops.UniformCandidateSampler

  380.     mindspore.ops.UniformInt

  381.     mindspore.ops.UniformReal

  382. 

  383. 图像算子

  384. ^^^^^^^^^^^^^^^

  385. 

  386. .. mscnplatformautosummary::

  387.     :toctree: ops

  388.     :nosignatures:

  389.     :template: classtemplate.rst

  390. 

  391.     mindspore.ops.CropAndResize

  392. 

  393. 稀疏算子

  394. ^^^^^^^^^^^^^^^^

  395. 

  396. .. mscnplatformautosummary::

  397.     :toctree: ops

  398.     :nosignatures:

  399.     :template: classtemplate.rst

  400. 

  401.     mindspore.ops.SparseToDense

  402.     mindspore.ops.SparseTensorDenseMatmul

  403. 

  404. 自定义算子

  405. ^^^^^^^^^^^^^^^^

  406. 

  407. .. mscnplatformautosummary::

  408.     :toctree: ops

  409.     :nosignatures:

  410.     :template: classtemplate.rst

  411. 

  412.     mindspore.ops.Custom

  413. 

  414. 其他算子

  415. ^^^^^^^^^^^^^^^

  416. 

  417. .. mscnplatformautosummary::

  418.     :toctree: ops

  419.     :nosignatures:

  420.     :template: classtemplate.rst

  421. 

  422.     mindspore.ops.Assign

  423.     mindspore.ops.BoundingBoxDecode

  424.     mindspore.ops.BoundingBoxEncode

  425.     mindspore.ops.CheckBprop

  426.     mindspore.ops.CheckValid

  427.     mindspore.ops.Depend

  428.     mindspore.ops.InTopK

  429.     mindspore.ops.IOU

  430.     mindspore.ops.NoRepeatNGram

  431.     mindspore.ops.Partial

  432.     mindspore.ops.PopulationCount