335 Star 1.5K Fork 863

MindSpore / docs

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
dynamic_shape_primitive.md 26.48 KB
一键复制 编辑 原始数据 按行查看 历史
李林杰 提交于 2023-09-01 10:40 . update dyn shape support table

Dynamic Shape Support Status of primitive Interface

View Source On Gitee

The following list provides primitive interfaces that support dynamic shape functionality in PYNATIVE mode. However, some primitive interfaces may have incomplete data type support. If you encounter such issues, you can resolve them by manually incorporating the Cast operator.

Primitive interfaces outside of this list have limited support for dynamic shape functionality and may fail to execute. Additionally, in graph mode, dynamic shape functionality is also limited and may result in execution failures.

If you encounter issues that the execution of dynamic shape operator fails, it is recommended to avoid introducing dynamic shape in the network. For example, you can adjust the inputs of primitive interfaces to construct a fully static shape network or confine dynamic shape to a local scope within the network.

Operator name Ascend GPU CPU
mindspore.Abs ✔️ ✔️ ✔️
mindspore.Acosh ✔️ ✔️ ✔️
mindspore.Adam ✔️ ✔️ ✔️
mindspore.AdaptiveAvgPool2D ✔️ ✔️ ✔️
mindspore.AdaptiveAvgPool3D ✔️ ✔️ ✔️
mindspore.Add ✔️ ✔️ ✔️
mindspore.Addcmul ✔️ ✔️ ✔️
mindspore.AddN ✔️ ✔️ ✔️
mindspore.Angle ✔️ ✔️ ✔️
mindspore.ArgMaxWithValue ✔️ ✔️ ✔️
mindspore.ArgMinWithValue ✔️ ✔️ ✔️
mindspore.Asin ✔️ ✔️ ✔️
mindspore.Asinh ✔️ ✔️ ✔️
mindspore.Assign ✔️ ✔️ ✔️
mindspore.AssignAdd ✔️ ✔️ ✔️
mindspore.Atan ✔️ ✔️ ✔️
mindspore.Atan2 ✔️ ✔️ ✔️
mindspore.Atanh ✔️ ✔️ ✔️
mindspore.AvgPool ✔️ ✔️ ✔️
mindspore.AvgPool3D ✔️ ✔️ ✔️
mindspore.BatchNorm ✔️ ✔️ ✔️
mindspore.BCEWithLogitsLoss ✔️ ✔️ ✔️
mindspore.Bernoulli ✔️ ✔️
mindspore.BesselI0 ✔️ ✔️
mindspore.BesselI0e ✔️ ✔️ ✔️
mindspore.BesselI1 ✔️ ✔️
mindspore.BesselI1e ✔️ ✔️ ✔️
mindspore.BesselJ0 ✔️ ✔️
mindspore.BesselJ1 ✔️ ✔️
mindspore.BiasAdd ✔️ ✔️ ️❌
mindspore.BinaryCrossEntropy ✔️ ✔️ ✔️
mindspore.BitwiseAnd ✔️ ✔️ ✔️
mindspore.BitwiseOr ✔️ ✔️ ✔️
mindspore.BitwiseXor ✔️ ✔️ ✔️
mindspore.BroadcastTo ✔️ ✔️ ✔️
mindspore.Cast ✔️ ✔️ ✔️
mindspore.Ceil ✔️ ✔️ ✔️
mindspore.Col2Im ✔️ ✔️ ✔️
mindspore.Complex ✔️ ✔️ ✔️
mindspore.Concat ✔️ ✔️ ✔️
mindspore.Conj ✔️ ✔️
mindspore.Conv2D ✔️ ✔️ ✔️
mindspore.Conv2DTranspose ✔️ ✔️ ✔️
mindspore.Conv3D ✔️ ✔️ ✔️
mindspore.Conv3DTranspose ✔️ ✔️ ✔️
mindspore.Cos ✔️ ✔️ ✔️
mindspore.Cosh ✔️ ✔️ ✔️
mindspore.Cross ✔️ ✔️
mindspore.CTCLossV2 ✔️ ✔️ ✔️
mindspore.Cummax ✔️ ✔️
mindspore.Cummin ✔️ ✔️ ✔️
mindspore.CumSum ✔️ ✔️ ️❌
mindspore.Diag ✔️ ✔️ ✔️
mindspore.Digamma ✔️ ✔️
mindspore.Div ✔️ ✔️ ✔️
mindspore.Dropout ✔️ ✔️ ✔️
mindspore.Dropout2D ✔️ ✔️ ✔️
mindspore.Dropout3D ✔️ ✔️ ✔️
mindspore.DynamicGRUV2 ✔️
mindspore.Einsum ✔️
mindspore.Elu ✔️ ✔️ ✔️
mindspore.Equal ✔️ ✔️ ✔️
mindspore.Erf ✔️ ✔️ ✔️
mindspore.Erfc ✔️ ✔️ ✔️
mindspore.Exp ✔️ ✔️ ✔️
mindspore.ExpandDims ✔️ ✔️ ✔️
mindspore.Expm1 ✔️ ✔️ ✔️
mindspore.Eye ✔️ ✔️ ✔️
mindspore.Fill ✔️ ✔️ ✔️
mindspore.FillV2 ✔️ ✔️ ✔️
mindspore.Flatten ✔️ ✔️ ✔️
mindspore.Floor ✔️ ✔️ ✔️
mindspore.FloorDiv ✔️ ✔️ ✔️
mindspore.FloorMod ✔️ ✔️ ✔️
mindspore.Gather ✔️ ✔️ ✔️
mindspore.GatherD ✔️ ✔️ ✔️
mindspore.GatherNd ✔️ ✔️ ✔️
mindspore.Gcd ✔️ ✔️ ✔️
mindspore.Geqrf ✔️ ✔️ ✔️
mindspore.Ger ✔️ ✔️ ✔️
mindspore.Greater ✔️ ✔️ ✔️
mindspore.GreaterEqual ✔️ ✔️ ✔️
mindspore.GridSampler2D ✔️ ✔️ ✔️
mindspore.GridSampler3D ✔️ ✔️ ✔️
mindspore.Heaviside ✔️ ✔️ ✔️
mindspore.HSwish ✔️ ✔️ ✔️
mindspore.Hypot ✔️ ✔️ ✔️
mindspore.Identity ✔️ ✔️ ✔️
mindspore.Imag ✔️ ✔️ ✔️
mindspore.Invert ✔️ ✔️ ✔️
mindspore.IsFinite ✔️ ✔️ ✔️
mindspore.IsInf ✔️ ✔️ ✔️
mindspore.IsNan ✔️ ✔️ ✔️
mindspore.KLDivLoss ✔️ ✔️ ✔️
mindspore.LayerNorm ✔️ ✔️ ✔️
mindspore.Lcm ✔️ ✔️ ✔️
mindspore.LeftShift ✔️ ✔️ ✔️
mindspore.Lerp ✔️ ✔️ ✔️
mindspore.Less ✔️ ✔️ ✔️
mindspore.LessEqual ✔️ ✔️ ✔️
mindspore.LinSpace ✔️ ✔️ ✔️
mindspore.Log ✔️ ✔️ ✔️
mindspore.Log1p ✔️ ✔️ ✔️
mindspore.LogicalAnd ✔️ ✔️ ✔️
mindspore.LogicalNot ✔️ ✔️ ✔️
mindspore.LogicalOr ✔️ ✔️ ✔️
mindspore.LogicalXor ✔️ ✔️
mindspore.Logit ✔️ ✔️ ✔️
mindspore.LogSoftmax ✔️ ✔️ ✔️
mindspore.LpNorm ✔️ ✔️ ✔️
mindspore.LRN ✔️ ✔️ ✔️
mindspore.MaskedFill ✔️ ✔️ ✔️
mindspore.MaskedSelect ✔️ ✔️ ✔️
mindspore.MatrixInverse ✔️ ✔️
mindspore.MatrixSolve ✔️ ✔️
mindspore.Maximum ✔️ ✔️ ✔️
mindspore.MaxPool ✔️ ✔️ ✔️
mindspore.MaxPool3D ✔️ ✔️ ✔️
mindspore.MaxUnpool2D ✔️ ✔️
mindspore.Median ✔️ ✔️
mindspore.Meshgrid ✔️ ✔️ ✔️
mindspore.Minimum ✔️ ✔️ ✔️
mindspore.MirrorPad ✔️ ✔️
mindspore.Mish ✔️ ✔️
mindspore.Mod ✔️ ✔️ ✔️
mindspore.Mul ✔️ ✔️ ✔️
mindspore.Multinomial ✔️ ✔️ ✔️
mindspore.Mvlgamma ✔️ ✔️ ✔️
mindspore.NanToNum ✔️ ✔️
mindspore.Neg ✔️ ✔️ ✔️
mindspore.NextAfter ✔️ ✔️ ✔️
mindspore.NLLLoss ✔️ ✔️ ✔️
mindspore.nonzero ✔️ ✔️ ✔️
mindspore.NotEqual ✔️ ✔️ ✔️
mindspore.OneHot ✔️ ✔️ ✔️
mindspore.OnesLike ✔️ ✔️ ✔️
mindspore.Pad ✔️ ✔️ ✔️
mindspore.Polar ✔️ ✔️
mindspore.Polygamma ✔️ ✔️
mindspore.Pow ✔️ ✔️ ✔️
mindspore.PReLU ✔️ ✔️ ✔️
mindspore.RandpermV2 ✔️
mindspore.Range ✔️ ✔️
mindspore.Real ✔️ ✔️
mindspore.RealDiv ✔️ ✔️ ✔️
mindspore.Reciprocal ✔️ ✔️ ✔️
mindspore.ReduceAll ✔️ ✔️ ✔️
mindspore.ReduceAny ✔️ ✔️ ✔️
mindspore.ReduceMax ✔️ ✔️ ✔️
mindspore.ReduceMean ✔️ ✔️ ✔️
mindspore.ReduceMin ✔️ ✔️ ✔️
mindspore.ReduceProd ✔️ ✔️ ✔️
mindspore.ReduceSum ✔️ ✔️ ✔️
mindspore.Reshape ✔️ ✔️ ✔️
mindspore.ResizeBilinearV2 ✔️ ✔️ ️❌
mindspore.ReverseSequence ✔️ ✔️ ✔️
mindspore.ReverseV2 ✔️ ✔️ ✔️
mindspore.RightShift ✔️ ✔️ ✔️
mindspore.Rint ✔️ ✔️ ✔️
mindspore.Round ✔️ ✔️ ✔️
mindspore.Rsqrt ✔️ ✔️ ✔️
mindspore.ScatterNd ✔️ ✔️ ✔️
mindspore.ScatterNdAdd ✔️ ✔️ ✔️
mindspore.ScatterNdMax ✔️ ✔️ ✔️
mindspore.ScatterNdMin ✔️ ✔️ ✔️
mindspore.ScatterNdMul ✔️ ✔️ ✔️
mindspore.ScatterNdSub ✔️ ✔️ ✔️
mindspore.ScatterNdUpdate ✔️ ✔️ ✔️
mindspore.ScatterSub ✔️ ✔️ ✔️
mindspore.ScatterUpdate ✔️ ✔️ ✔️
mindspore.Select ✔️ ✔️ ✔️
mindspore.Sigmoid ✔️ ✔️ ✔️
mindspore.Sign ✔️ ✔️ ✔️
mindspore.Sin ✔️ ✔️ ✔️
mindspore.Sinc ✔️ ✔️ ✔️
mindspore.Sinh ✔️ ✔️ ✔️
mindspore.Slice ✔️ ✔️ ✔️
mindspore.SmoothL1Loss ✔️ ✔️ ✔️
mindspore.Softmax ✔️ ✔️ ✔️
mindspore.SoftmaxCrossEntropyWithLogits ✔️ ✔️ ✔️
mindspore.Softplus ✔️ ✔️ ✔️
mindspore.SoftShrink ✔️ ✔️ ✔️
mindspore.Sort ✔️ ✔️
mindspore.Split ✔️ ✔️ ️❌
mindspore.Sqrt ✔️ ✔️ ✔️
mindspore.Square ✔️ ✔️ ✔️
mindspore.Squeeze ✔️ ✔️ ✔️
mindspore.Stack ✔️ ✔️ ✔️
mindspore.StandardNormal ✔️ ✔️ ✔️
mindspore.StridedSlice ✔️ ✔️ ✔️
mindspore.Sub ✔️ ✔️ ✔️
mindspore.Svd ✔️ ✔️
mindspore.Tan ✔️ ✔️ ✔️
mindspore.Tanh ✔️ ✔️ ✔️
mindspore.Tile ✔️ ✔️ ✔️
mindspore.TopK ✔️ ✔️ ✔️
mindspore.Trace ✔️ ✔️ ✔️
mindspore.Transpose ✔️ ✔️ ✔️
mindspore.Tril ✔️ ✔️ ✔️
mindspore.Trunc ✔️ ✔️ ✔️
mindspore.TruncateDiv ✔️ ✔️ ️❌
mindspore.UniformInt ✔️ ✔️
mindspore.UniformReal ✔️ ✔️
mindspore.Unique ✔️ ✔️ ✔️
mindspore.UnsortedSegmentSum ✔️ ✔️ ✔️
mindspore.Xlogy ✔️ ✔️ ✔️
mindspore.ZerosLike ✔️ ✔️ ✔️
1
https://gitee.com/mindspore/docs.git
git@gitee.com:mindspore/docs.git
mindspore
docs
docs
r2.0

搜索帮助

53164aa7 5694891 3bd8fe86 5694891