算子支持

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	Doing	layer/activation
mindspore.nn.HSwish	Doing	Supported	Doing	layer/activation
mindspore.nn.HSigmoid	Doing	Supported	Doing	layer/activation
mindspore.nn.LeakyReLU	Supported	Doing	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	Doing	layer/basic
mindspore.nn.Flatten	Supported	Supported	Doing	layer/basic
mindspore.nn.Dense	Supported	Supported	Doing	layer/basic
mindspore.nn.DenseBnAct	Supported	Doing	Supported	layer/basic
mindspore.nn.ClipByNorm	Doing	Supported	Doing	layer/basic
mindspore.nn.Norm	Doing	Supported	Doing	layer/basic
mindspore.nn.OneHot	Supported	Supported	Doing	layer/basic
mindspore.nn.Range	Supported	Doing	Doing	layer/basic
mindspore.nn.SequentialCell	Doing	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.Embedding	Doing	Supported	Doing	layer/embedding
mindspore.nn.ImageGradients	Doing	Doing	Doing	layer/image
mindspore.nn.SSIM	Doing	Doing	Doing	layer/image
mindspore.nn.PSNR	Doing	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.LinSpace	Supported	Doing	Doing	layer/normalization
mindspore.nn.MaxPool2d	Supported	Supported	Supported	layer/pooling
mindspore.nn.AvgPool2d	Doing	Supported	Doing	layer/pooling
mindspore.nn.L1Loss	Doing	Doing	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	Doing	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.AdamWeightDecayDynamicLR	Supported	Supported	Doing	optim/adam
mindspore.nn.Lamb	Supported	Supported	Doing	optim/lamb
mindspore.nn.LARS	Doing	Doing	Doing	optim/lars
mindspore.nn.Momentum	Supported	Supported	Doing	optim/momentum
mindspore.nn.Optimizer	Supported	Supported	Doing	optim/optimizer
mindspore.nn.RMSProp	Supported	Supported	Doing	optim/optimizer
mindspore.nn.SGD	Supported	Doing	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	Doing	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	Doing	Doing	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	Doing	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	Doing	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.Conv2D	Supported	Supported	Supported	nn_ops
mindspore.ops.operations.DepthwiseConv2dNative	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	Doing	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.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.SparseApplyProximalAdagrad	Supported	Doing	Doing	nn_ops
mindspore.ops.operations.ApplyProximalAdagrad	Supported	Doing	Doing	nn_ops
mindspore.ops.operations.SparseApplyLazyAdam	Doing	Doing	Supported	nn_ops
mindspore.ops.operations.SparseApplyAdam	Doing	Doing	Supported	nn_ops
mindspore.ops.operations.SmoothL1Loss	Supported	Doing	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	Doing	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	Doing	Doing	nn_ops
mindspore.ops.operations.ROIAlign	Supported	Doing	Doing	nn_ops
mindspore.ops.operations.Adam	Supported	Supported	Doing	nn_ops
mindspore.ops.operations.BinaryCrossEntropy	Supported	Doing	Doing	nn_ops
mindspore.ops.operations.LARSUpdate	Supported	Doing	Doing	nn_ops
mindspore.ops.operations.TensorAdd	Supported	Supported	Doing	math_ops
mindspore.ops.operations.AssignAdd	Supported	Supported	Doing	math_ops
mindspore.ops.operations.AssignSub	Supported	Doing	Doing	math_ops
mindspore.ops.operations.ReduceMean	Supported	Supported	Doing	math_ops
mindspore.ops.operations.ReduceSum	Supported	Supported	Doing	math_ops
mindspore.ops.operations.ReduceAll	Supported	Doing	Doing	math_ops
mindspore.ops.operations.ReduceMax	Supported	Supported	Doing	math_ops
mindspore.ops.operations.ReduceMin	Supported	Doing	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	Doing	Doing	math_ops
mindspore.ops.operations.AddN	Supported	Supported	Doing	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	Supported	Doing	math_ops
mindspore.ops.operations.Sqrt	Supported	Supported	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	Doing	Doing	math_ops
mindspore.ops.operations.DivNoNan	Supported	Doing	Doing	math_ops
mindspore.ops.operations.FloorDiv	Supported	Doing	Doing	math_ops
mindspore.ops.operations.Floor	Supported	Doing	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.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	Doing	Doing	math_ops
mindspore.ops.operations.Abs	Supported	Doing	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.ExpandDims	Supported	Supported	Doing	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	Doing	array_ops
mindspore.ops.operations.GatherV2	Supported	Supported	Doing	array_ops
mindspore.ops.operations.Split	Supported	Doing	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	Doing	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	Doing	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.Concat	Supported	Supported	Doing	array_ops
mindspore.ops.operations.Slice	Supported	Supported	Doing	array_ops
mindspore.ops.operations.Select	Supported	Supported	Doing	array_ops
mindspore.ops.operations.StridedSlice	Supported	Supported	Doing	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	Doing	Doing	array_ops
mindspore.ops.operations.ResizeNearestNeighbor	Supported	Doing	Doing	array_ops
mindspore.ops.operations.GatherNd	Supported	Doing	Doing	array_ops
mindspore.ops.operations.ApplyFtrl	Supported	Supported	Doing	array_ops
mindspore.ops.operations.SparseApplyFtrl	Supported	Doing	Doing	array_ops
mindspore.ops.operations.ScatterNdUpdate	Supported	Doing	Doing	array_ops
mindspore.ops.operations.ScatterUpdate	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.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.HostAllGather	Doing	Doing	Supported	comm_ops
mindspore.ops.operations.ReduceScatter	Doing	Supported	Doing	comm_ops
mindspore.ops.operations.HostReduceScatter	Doing	Doing	Supported	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	Doing	Doing	other_ops
mindspore.ops.operations.BoundingBoxDecode	Supported	Doing	Doing	other_ops
mindspore.ops.operations.CheckValid	Supported	Doing	Doing	other_ops
mindspore.ops.operations.IOU	Supported	Doing	Doing	other_ops
mindspore.ops.operations.MakeRefKey	Supported	Supported	Supported	other_ops
mindspore.ops.operations.InTopK	Supported	Doing	Doing	other_ops
mindspore.ops.operations.RandomChoiceWithMask	Doing	Doing	Doing	random_ops
mindspore.ops.operations.ScalarCast	Supported	Supported	Supported	inner_ops