算子支持

Ascend GPU CPU 模型开发 初级 中级 高级

查看源文件

mindspore.nn

操作名

Ascend

GPU

CPU

算子类别

mindspore.nn.Softmax

Supported

Supported

Supported

layer/activation

mindspore.nn.LogSoftmax

Supported

Supported

Doing

layer/activation

mindspore.nn.ReLU

Supported

Supported

Supported

layer/activation

mindspore.nn.ReLU6

Supported

Supported

Supported

layer/activation

mindspore.nn.HSwish

Doing

Supported

Doing

layer/activation

mindspore.nn.HSigmoid

Doing

Supported

Doing

layer/activation

mindspore.nn.LeakyReLU

Supported

Supported

Doing

layer/activation

mindspore.nn.Tanh

Supported

Supported

Doing

layer/activation

mindspore.nn.GELU

Supported

Supported

Doing

layer/activation

mindspore.nn.Sigmoid

Supported

Supported

Doing

layer/activation

mindspore.nn.PReLU

Supported

Doing

Doing

layer/activation

mindspore.nn.Dropout

Supported

Supported

Supported

layer/basic

mindspore.nn.Flatten

Supported

Supported

Supported

layer/basic

mindspore.nn.Dense

Supported

Supported

Supported

layer/basic

mindspore.nn.DenseBnAct

Supported

Doing

Supported

layer/basic

mindspore.nn.ClipByNorm

Supported

Supported

Doing

layer/basic

mindspore.nn.Norm

Doing

Supported

Doing

layer/basic

mindspore.nn.OneHot

Supported

Supported

Supported

layer/basic

mindspore.nn.Range

Supported

Doing

Doing

layer/basic

mindspore.nn.SequentialCell

Supported

Supported

Doing

layer/container

mindspore.nn.CellList

Supported

Supported

Doing

layer/container

mindspore.nn.Conv2d

Supported

Supported

Supported

layer/conv

mindspore.nn.Conv2dTranspose

Supported

Supported

Doing

layer/conv

mindspore.nn.Conv2dBnAct

Supported

Supported

Supported

layer/conv

mindspore.nn.Conv1d

Supported

Supported

Doing

layer/conv

mindspore.nn.Conv1dTranspose

Supported

Supported

Doing

layer/conv

mindspore.nn.Embedding

Supported

Supported

Doing

layer/embedding

mindspore.nn.ImageGradients

Doing

Doing

Doing

layer/image

mindspore.nn.SSIM

Supported

Supported

Doing

layer/image

mindspore.nn.PSNR

Supported

Doing

Doing

layer/image

mindspore.nn.CentralCrop

Supported

Doing

Doing

layer/image

mindspore.nn.LSTM

Doing

Supported

Supported

layer/lstm

mindspore.nn.GlobalBatchNorm

Supported

Doing

Doing

layer/normalization

mindspore.nn.BatchNorm1d

Supported

Doing

Doing

layer/normalization

mindspore.nn.BatchNorm2d

Supported

Supported

Doing

layer/normalization

mindspore.nn.GroupNorm

Supported

Doing

Doing

layer/normalization

mindspore.nn.LayerNorm

Supported

Supported

Doing

layer/normalization

mindspore.nn.MatrixDiag

Supported

Doing

Doing

layer/normalization

mindspore.nn.MatrixDiagPart

Supported

Doing

Doing

layer/normalization

mindspore.nn.MatrixSetDiag

Supported

Doing

Doing

layer/normalization

mindspore.nn.LinSpace

Supported

Doing

Doing

layer/normalization

mindspore.nn.MaxPool2d

Supported

Supported

Supported

layer/pooling

mindspore.nn.AvgPool2d

Supported

Supported

Doing

layer/pooling

mindspore.nn.L1Loss

Supported

Supported

Doing

loss/loss

mindspore.nn.MSELoss

Supported

Doing

Doing

loss/loss

mindspore.nn.SmoothL1Loss

Supported

Doing

Doing

loss/loss

mindspore.nn.SoftmaxCrossEntropyWithLogits

Supported

Supported

Doing

loss/loss

mindspore.nn.SoftmaxCrossEntropyExpand

Supported

Supported

Doing

loss/loss

mindspore.nn.CosineEmbeddingLoss

Supported

Supported

Doing

loss/loss

mindspore.nn.ProximalAdagrad

Supported

Doing

Doing

optim/ProximalAdagrad

mindspore.nn.LazyAdam

Supported

Doing

Doing

optim/lazyadam

mindspore.nn.Adam

Supported

Doing

Doing

optim/adam

mindspore.nn.AdamWeightDecay

Supported

Supported

Doing

optim/adam

mindspore.nn.Lamb

Supported

Supported

Doing

optim/lamb

mindspore.nn.LARS

Supported

Doing

Doing

optim/lars

mindspore.nn.Momentum

Supported

Supported

Supported

optim/momentum

mindspore.nn.Optimizer

Supported

Supported

Doing

optim/optimizer

mindspore.nn.RMSProp

Supported

Supported

Doing

optim/optimizer

mindspore.nn.SGD

Supported

Supported

Doing

optim/sgd

mindspore.nn.WithLossCell

Supported

Supported

Doing

wrap/cell_wrapper

mindspore.nn.WithGradCell

Supported

Supported

Doing

wrap/cell_wrapper

mindspore.nn.TrainOneStepCell

Supported

Supported

Doing

wrap/cell_wrapper

mindspore.nn.DataWrapper

Doing

Supported

Doing

wrap/cell_wrapper

mindspore.nn.GetNextSingleOp

Doing

Supported

Doing

wrap/cell_wrapper

mindspore.nn.WithEvalCell

Supported

Supported

Doing

wrap/cell_wrapper

mindspore.nn.ParameterUpdate

Supported

Doing

Doing

wrap/cell_wrapper

mindspore.nn.DistributedGradReducer

Supported

Doing

Doing

wrap/grad_reducer

mindspore.nn.DynamicLossScaleUpdateCell

Doing

Doing

Doing

wrap/loss_scale

mindspore.nn.FixedLossScaleUpdateCell

Doing

Doing

Doing

wrap/loss_scale

mindspore.nn.TrainOneStepWithLossScaleCell

Doing

Doing

Doing

wrap/loss_scale

mindspore.nn.Cell

Supported

Supported

Supported

cell

mindspore.ops.operations

操作名

Ascend

GPU

CPU

算子类别

mindspore.ops.operations.Flatten

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.Softmax

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.Acosh

Doing

Doing

Doing

nn_ops

mindspore.ops.operations.FloorMod

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.Elu

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.MirrorPad

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.Unpack

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.Pack

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.L2Loss

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.CTCLoss

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.RNNTLoss

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.LogSoftmax

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.Softplus

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ReLU

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.ReLU6

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.HSwish

Doing

Supported

Doing

nn_ops

mindspore.ops.operations.HSigmoid

Doing

Supported

Doing

nn_ops

mindspore.ops.operations.Sigmoid

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.Tanh

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.BatchNorm

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.LRN

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.Conv2D

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.DepthwiseConv2dNative

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.DepthwiseConv2dNativeBackpropInput

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.DepthwiseConv2dNativeiBackpropFilter

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.MaxPoolWithArgmax

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.MaxPool

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.AvgPool

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.Conv2DBackpropInput

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.BiasAdd

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.TopK

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.SoftmaxCrossEntropyWithLogits

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.SparseSoftmaxCrossEntropyWithLogits

Doing

Supported

Supported

nn_ops

mindspore.ops.operations.ApplyMomentum

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.ApplyAddSign

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ApplyPowerSign

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ApplyGradientDescent

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ApplyProximalGradientDescent

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ApplyRMSProp

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.ApplyCenteredRMSProp

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.SparseApplyAdagrad

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.SparseApplyAdagradV2

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.SparseApplyProximalAdagrad

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.FusedSparseProximalAdagrad

Doing

Doing

Supported

nn_ops

mindspore.ops.operations.ApplyProximalAdagrad

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.FusedSparseLazyAdam

Doing

Doing

Supported

nn_ops

mindspore.ops.operations.FusedSparseAdam

Doing

Doing

Supported

nn_ops

mindspore.ops.operations.SmoothL1Loss

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.SGD

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.LayerNorm

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.L2Normalize

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.DropoutGenMask

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.DropoutDoMask

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ResizeBilinear

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.OneHot

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.Gelu

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.GetNext

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.PReLU

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.LSTM

Doing

Supported

Supported

nn_ops

mindspore.ops.operations.BasicLSTMCell

Doing

Doing

Doing

nn_ops

mindspore.ops.operations.SigmoidCrossEntropyWithLogits

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.Pad

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.ROIAlign

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.Adam

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.BinaryCrossEntropy

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.KLDivLoss

Doing

Supported

Doing

nn_ops

mindspore.ops.operations.LARSUpdate

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.Softsign

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.TensorAdd

Supported

Supported

Supported

math_ops

mindspore.ops.operations.AssignAdd

Supported

Supported

Supported

math_ops

mindspore.ops.operations.AssignSub

Supported

Doing

Doing

math_ops

mindspore.ops.operations.ReduceMean

Supported

Supported

Supported

math_ops

mindspore.ops.operations.ReduceSum

Supported

Supported

Supported

math_ops

mindspore.ops.operations.ReduceAll

Supported

Doing

Doing

math_ops

mindspore.ops.operations.ReduceMax

Supported

Supported

Supported

math_ops

mindspore.ops.operations.ReduceMin

Supported

Supported

Doing

math_ops

mindspore.ops.operations.ReduceProd

Supported

Doing

Doing

math_ops

mindspore.ops.operations.CumProd

Supported

Doing

Doing

math_ops

mindspore.ops.operations.MatMul

Supported

Supported

Supported

math_ops

mindspore.ops.operations.BatchMatMul

Supported

Supported

Doing

math_ops

mindspore.ops.operations.CumSum

Supported

Supported

Doing

math_ops

mindspore.ops.operations.AddN

Supported

Supported

Supported

math_ops

mindspore.ops.operations.Neg

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Sub

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Mul

Supported

Supported

Supported

math_ops

mindspore.ops.operations.Square

Supported

Supported

Doing

math_ops

mindspore.ops.operations.SquareSumAll

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Rsqrt

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Sqrt

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Reciprocal

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Pow

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Exp

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Log

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Log1p

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Minimum

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Maximum

Supported

Supported

Doing

math_ops

mindspore.ops.operations.RealDiv

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Div

Supported

Supported

Doing

math_ops

mindspore.ops.operations.DivNoNan

Supported

Doing

Doing

math_ops

mindspore.ops.operations.FloorDiv

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Floor

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Equal

Supported

Supported

Doing

math_ops

mindspore.ops.operations.EqualCount

Doing

Supported

Supported

math_ops

mindspore.ops.operations.NotEqual

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Greater

Supported

Supported

Doing

math_ops

mindspore.ops.operations.GreaterEqual

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Less

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Atan2

Supported

Doing

Doing

math_ops

mindspore.ops.operations.LessEqual

Supported

Supported

Doing

math_ops

mindspore.ops.operations.LogicalNot

Supported

Supported

Doing

math_ops

mindspore.ops.operations.LogicalAnd

Supported

Supported

Doing

math_ops

mindspore.ops.operations.LogicalOr

Supported

Supported

Doing

math_ops

mindspore.ops.operations.BitwiseAnd

Supported

Doing

Doing

math_ops

mindspore.ops.operations.BitwiseOr

Supported

Doing

Doing

math_ops

mindspore.ops.operations.BitwiseXor

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Ceil

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Inv

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Invert

Supported

Doing

Doing

math_ops

mindspore.ops.operations.NPUAllocFloatStatus

Supported

Doing

Doing

math_ops

mindspore.ops.operations.NPUGetFloatStatus

Supported

Doing

Doing

math_ops

mindspore.ops.operations.NPUClearFloatStatus

Supported

Doing

Doing

math_ops

mindspore.ops.operations.FloatStatus

Doing

Supported

Doing

math_ops

mindspore.ops.operations.Cos

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Cosh

Supported

Doing

Doing

math_ops

mindspore.ops.operations.ACos

Supported

Doing

Doing

math_ops

mindspore.ops.operations.BesselI0e

Supported

Doing

Doing

math_ops

mindspore.ops.operations.BesselI1e

Supported

Doing

Doing

math_ops

mindspore.ops.operations.TruncateDiv

Supported

Doing

Doing

math_ops

mindspore.ops.operations.TruncateMod

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Tan

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Asin

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Asinh

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Erf

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Erfc

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Sin

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Sinh

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Expm1

Supported

Doing

Doing

math_ops

mindspore.ops.operations.NMSWithMask

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Abs

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Sign

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Round

Supported

Doing

Doing

math_ops

mindspore.ops.operations.ApproximateEqual

Supported

Doing

Doing

math_ops

mindspore.ops.operations.InplaceAdd

Supported

Doing

Doing

math_ops

mindspore.ops.operations.InplaceSub

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Mod

Supported

Doing

Doing

math_ops

mindspore.ops.operations.ExpandDims

Supported

Supported

Supported

array_ops

mindspore.ops.operations.DType

Supported

Supported

Supported

array_ops

mindspore.ops.operations.SameTypeShape

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Cast

Supported

Supported

Doing

array_ops

mindspore.ops.operations.IsSubClass

Supported

Supported

Supported

array_ops

mindspore.ops.operations.IsInstance

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Reshape

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Shape

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Squeeze

Supported

Supported

Doing

array_ops

mindspore.ops.operations.Transpose

Supported

Supported

Supported

array_ops

mindspore.ops.operations.GatherV2

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Split

Supported

Supported

Doing

array_ops

mindspore.ops.operations.Rank

Supported

Supported

Supported

array_ops

mindspore.ops.operations.TruncatedNormal

Doing

Doing

Doing

array_ops

mindspore.ops.operations.Size

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Fill

Supported

Supported

Supported

array_ops

mindspore.ops.operations.OnesLike

Supported

Supported

Doing

array_ops

mindspore.ops.operations.ZerosLike

Supported

Supported

Doing

array_ops

mindspore.ops.operations.TupleToArray

Supported

Supported

Supported

array_ops

mindspore.ops.operations.ScalarToArray

Supported

Supported

Supported

array_ops

mindspore.ops.operations.ScalarToTensor

Supported

Supported

Supported

array_ops

mindspore.ops.operations.InvertPermutation

Supported

Supported

Doing

array_ops

mindspore.ops.operations.Argmax

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Argmin

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ArgMaxWithValue

Supported

Supported

Doing

array_ops

mindspore.ops.operations.ArgMinWithValue

Supported

Doing

Doing

array_ops

mindspore.ops.operations.Tile

Supported

Supported

Doing

array_ops

mindspore.ops.operations.UnsortedSegmentSum

Supported

Supported

Doing

array_ops

mindspore.ops.operations.UnsortedSegmentMin

Supported

Doing

Doing

array_ops

mindspore.ops.operations.UnsortedSegmentProd

Supported

Doing

Doing

array_ops

mindspore.ops.operations.Concat

Supported

Supported

Supported

array_ops

mindspore.ops.operations.ParallelConcat

Supported

Doing

Doing

array_ops

mindspore.ops.operations.Slice

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Select

Supported

Supported

Doing

array_ops

mindspore.ops.operations.StridedSlice

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Diag

Doing

Doing

Doing

array_ops

mindspore.ops.operations.DiagPart

Doing

Doing

Doing

array_ops

mindspore.ops.operations.Eye

Supported

Supported

Supported

array_ops

mindspore.ops.operations.ScatterNd

Supported

Supported

Doing

array_ops

mindspore.ops.operations.ResizeNearestNeighbor

Supported

Supported

Doing

array_ops

mindspore.ops.operations.GatherNd

Supported

Supported

Doing

array_ops

mindspore.ops.operations.ApplyFtrl

Supported

Supported

Doing

array_ops

mindspore.ops.operations.SparseApplyFtrl

Supported

Doing

Doing

array_ops

mindspore.ops.operations.FusedSparseFtrl

Doing

Doing

Supported

array_ops

mindspore.ops.operations.SparseApplyFtrlV2

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterNdUpdate

Supported

Doing

Supported

array_ops

mindspore.ops.operations.ScatterUpdate

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterMul

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterDiv

Supported

Doing

Doing

array_ops

mindspore.ops.operations.SpaceToDepth

Supported

Doing

Doing

array_ops

mindspore.ops.operations.DepthToSpace

Supported

Doing

Doing

array_ops

mindspore.ops.operations.SpaceToBatch

Supported

Doing

Doing

array_ops

mindspore.ops.operations.SpaceToBatchND

Supported

Doing

Doing

array_ops

mindspore.ops.operations.BatchToSpace

Supported

Doing

Doing

array_ops

mindspore.ops.operations.BatchToSpaceND

Supported

Doing

Doing

array_ops

mindspore.ops.operations.IsFinite

Supported

Supported

Doing

array_ops

mindspore.ops.operations.InplaceUpdate

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterSub

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterMax

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterMin

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterNdAdd

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterNdSub

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterNonAliasingAdd

Supported

Doing

Doing

array_ops

mindspore.ops.operations.Rint

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ReverseV2

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ReduceOp

Supported

Supported

Doing

comm_ops

mindspore.ops.operations.AllReduce

Supported

Supported

Doing

comm_ops

mindspore.ops.operations.AllGather

Supported

Supported

Doing

comm_ops

mindspore.ops.operations.ReduceScatter

Doing

Supported

Doing

comm_ops

mindspore.ops.operations.Broadcast

Supported

Doing

Doing

comm_ops

mindspore.ops.operations.ControlDepend

Supported

Supported

Supported

control_ops

mindspore.ops.operations.GeSwitch

Doing

Doing

Doing

control_ops

mindspore.ops.operations.Merge

Doing

Doing

Doing

control_ops

mindspore.ops.operations.ScalarSummary

Supported

Supported

Supported

debug_ops

mindspore.ops.operations.ImageSummary

Supported

Supported

Supported

debug_ops

mindspore.ops.operations.TensorSummary

Supported

Supported

Supported

debug_ops

mindspore.ops.operations.HistogramSummary

Supported

Supported

Supported

debug_ops

mindspore.ops.operations.InsertGradientOf

Supported

Supported

Supported

debug_ops

mindspore.ops.operations.Print

Supported

Doing

Doing

debug_ops

mindspore.ops.operations.Assign

Supported

Supported

Doing

other_ops

mindspore.ops.operations.BoundingBoxEncode

Supported

Supported

Doing

other_ops

mindspore.ops.operations.BoundingBoxDecode

Supported

Supported

Doing

other_ops

mindspore.ops.operations.PopulationCount

Supported

Doing

Doing

other_ops

mindspore.ops.operations.CheckValid

Supported

Supported

Doing

other_ops

mindspore.ops.operations.IOU

Supported

Supported

Doing

other_ops

mindspore.ops.operations.MakeRefKey

Supported

Supported

Supported

other_ops

mindspore.ops.operations.InTopK

Supported

Doing

Doing

other_ops

mindspore.ops.operations.StandardNormal

Supported

Supported

Doing

random_ops

mindspore.ops.operations.Gamma

Supported

Doing

Doing

random_ops

mindspore.ops.operations.Poisson

Supported

Doing

Doing

random_ops

mindspore.ops.operations.UniformInt

Supported

Supported

Doing

random_ops

mindspore.ops.operations.UniformReal

Supported

Supported

Doing

random_ops

mindspore.ops.operations.RandomChoiceWithMask

Doing

Supported

Doing

random_ops

mindspore.ops.operations.RandomCategorical

Supported

Doing

Doing

random_ops

mindspore.ops.operations.ScalarCast

Supported

Supported

Supported

inner_ops

mindspore.ops.operations.ReverseSequence

Supported

Doing

Doing

array_ops

mindspore.ops.operations.CropAndResize

Supported

Doing

Doing

image_ops

mindspore.ops.operations.SquaredDifference

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Xdivy

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Xlogy

Supported

Doing

Doing

math_ops

mindspore.ops.functional

操作名

对应functional算子

mindspore.ops.operations.Pack

pack

mindspore.ops.operations.TensorAdd

tensor_add

mindspore.ops.operations.AssignSub

assign_sub

mindspore.ops.operations.AddN

addn

mindspore.ops.operations.Square

square

mindspore.ops.operations.Sqrt

sqrt

mindspore.ops.operations.Equal

equal

mindspore.ops.operations.NotEqual

not_equal

mindspore.ops.operations.LogicalNot

logical_not

mindspore.ops.operations.LogicalAnd

logical_and

mindspore.ops.operations.LogicalOr

logical_or

mindspore.ops.operations.ExpandDims

expand_dims

mindspore.ops.operations.DType

dtype

mindspore.ops.operations.Cast

cast

mindspore.ops.operations.Reshape

reshape

mindspore.ops.operations.Shape

shape

mindspore.ops.operations.GatherV2

gather

mindspore.ops.operations.Rank

rank

mindspore.ops.operations.Size

size

mindspore.ops.operations.Fill

fill

mindspore.ops.operations.OnesLike

ones_like

mindspore.ops.operations.Tile

tile

mindspore.ops.operations.Select

select

mindspore.ops.operations.ScatterNd

scatter_nd

mindspore.ops.operations.GatherNd

gather_nd

mindspore.ops.operations.ControlDepend

control_depend

mindspore.ops.operations.Print

print

mindspore.ops.operations.Assign

assign

mindspore.ops.operations.Pow

tensor_pow

当前functional支持了一部分没有属性的算子,后续会进一步补齐完整。

隐式类型转换

转换规则

  • 标量与Tensor运算:运算时,将标量自动转为Tensor,数据类型和参与运算的Tensor数据类型保持一致; 而当Tensor是bool数据类型,标量是int或float时,将标量和Tensor都转为数据类型为int32或float32的Tensor。

  • 不同数据类型Tensor运算:数据类型优先级排序为bool < uint8 < int8 < int16 < int32 < int64 < float16 < float32 < float64, 运算时,先确定参与运算的Tensor中优先级相对最高的数据类型,然后将低优先级数据类型Tensor转换为相对最高优先级数据类型; 而当int8和uint8数据类型的Tensor进行运算时,将其都转为int16的Tensor。

  • 不支持对Parameter进行数据类型转换:如果按照转换规则推导,需要对网络中定义的Parameter进行数据类型转换时,会抛出RuntimeError异常。

参与转换的数据类型

  • bool

  • int8

  • uint8

  • int16

  • int32

  • int64

  • float16

  • float32

  • float64

支持算子

算子名

mindspore.ops.operations.Assign

mindspore.ops.operations.AssignSub

mindspore.ops.operations.ApplyMomentum

mindspore.ops.operations.FusedSparseAdam

mindspore.ops.operations.FusedSparseLazyAdam

mindspore.ops.operations.FusedSparseFtrl

mindspore.ops.operations.FusedSparseProximalAdagrad

mindspore.ops.operations.ApplyAdaMax

mindspore.ops.operations.ApplyAdadelta

mindspore.ops.operations.ApplyAdagrad

mindspore.ops.operations.ApplyAdagradV2

mindspore.ops.operations.SparseApplyAdagrad

mindspore.ops.operations.SparseApplyAdagradV2

mindspore.ops.operations.ApplyProximalAdagrad

mindspore.ops.operations.SparseApplyProximalAdagrad

mindspore.ops.operations.ApplyAddSign

mindspore.ops.operations.ApplyPowerSign

mindspore.ops.operations.ApplyGradientDescent

mindspore.ops.operations.ApplyProximalGradientDescent

mindspore.ops.operations.SparseApplyFtrl

mindspore.ops.operations.SparseApplyFtrlV2

mindspore.ops.operations.BitwiseAnd

mindspore.ops.operations.BitwiseOr

mindspore.ops.operations.BitwiseXor

mindspore.ops.operations.TensorAdd

mindspore.ops.operations.Sub

mindspore.ops.operations.Mul

mindspore.ops.operations.Pow

mindspore.ops.operations.Minimum

mindspore.ops.operations.Maximum

mindspore.ops.operations.RealDiv

mindspore.ops.operations.Div

mindspore.ops.operations.DivNoNan

mindspore.ops.operations.FloorDiv

mindspore.ops.operations.TruncateDiv

mindspore.ops.operations.TruncateMod

mindspore.ops.operations.Mod

mindspore.ops.operations.FloorMod

mindspore.ops.operations.Atan2

mindspore.ops.operations.SquaredDifference

mindspore.ops.operations.Xdivy

mindspore.ops.operations.Xlogy

mindspore.ops.operations.Equal

mindspore.ops.operations.ApproximateEqual

mindspore.ops.operations.NotEqual

mindspore.ops.operations.Greater

mindspore.ops.operations.GreaterEqual

mindspore.ops.operations.Less

mindspore.ops.operations.LessEqual

mindspore.ops.operations.LogicalAnd

mindspore.ops.operations.LogicalOr

mindspore.ops.operations.ScatterNdUpdate

mindspore.ops.operations.ScatterNdAdd

mindspore.ops.operations.ScatterNdSub

mindspore.ops.operations.ScatterNonAliasingAdd

mindspore.ops.operations.ScatterUpdate

mindspore.ops.operations.ScatterMax

mindspore.ops.operations.ScatterMin

mindspore.ops.operations.ScatterAdd

mindspore.ops.operations.ScatterSub

mindspore.ops.operations.ScatterMul

mindspore.ops.operations.ScatterDiv