forked from mindspore-Ecosystem/mindspore
!45688 expose Primitve APIs for part3 and part4
Merge pull request !45688 from 李林杰/expose_Primitive_APIs_part3_part4
This commit is contained in:
Gitee
commit
3eea27abbe
|
|
@ -74,6 +74,8 @@ MindSpore中 `mindspore.ops` 接口与上一版本相比，新增、删除和支
|
|||
mindspore.ops.Padding
|
||||
mindspore.ops.ResizeNearestNeighbor
|
||||
mindspore.ops.ResizeBilinear
|
||||
mindspore.ops.UpsampleNearest3D
|
||||
mindspore.ops.UpsampleTrilinear3D
|
||||
|
||||
损失函数
|
||||
^^^^^^^^^^
|
||||
|
|
@ -96,6 +98,7 @@ MindSpore中 `mindspore.ops` 接口与上一版本相比，新增、删除和支
|
|||
mindspore.ops.SoftMarginLoss
|
||||
mindspore.ops.SoftmaxCrossEntropyWithLogits
|
||||
mindspore.ops.SparseSoftmaxCrossEntropyWithLogits
|
||||
mindspore.ops.TripletMarginLoss
|
||||
|
||||
激活函数
|
||||
^^^^^^^^^^
|
||||
|
|
@ -183,6 +186,7 @@ MindSpore中 `mindspore.ops` 接口与上一版本相比，新增、删除和支
|
|||
mindspore.ops.ComputeAccidentalHits
|
||||
mindspore.ops.LogUniformCandidateSampler
|
||||
mindspore.ops.UniformCandidateSampler
|
||||
mindspore.ops.UpsampleNearest3D
|
||||
|
||||
图像处理
|
||||
^^^^^^^^^^
|
||||
|
|
@ -203,7 +207,10 @@ MindSpore中 `mindspore.ops` 接口与上一版本相比，新增、删除和支
|
|||
mindspore.ops.IOU
|
||||
mindspore.ops.L2Normalize
|
||||
mindspore.ops.NMSWithMask
|
||||
mindspore.ops.RGBToHSV
|
||||
mindspore.ops.ROIAlign
|
||||
mindspore.ops.SampleDistortedBoundingBoxV2
|
||||
mindspore.ops.ScaleAndTranslate
|
||||
|
||||
文本处理
|
||||
^^^^^^^^^^
|
||||
|
|
@ -318,6 +325,7 @@ MindSpore中 `mindspore.ops` 接口与上一版本相比，新增、删除和支
|
|||
mindspore.ops.TruncateMod
|
||||
mindspore.ops.Xdivy
|
||||
mindspore.ops.Xlogy
|
||||
mindspore.ops.Zeta
|
||||
|
||||
|
||||
Reduction算子
|
||||
|
|
@ -378,6 +386,7 @@ Reduction算子
|
|||
mindspore.ops.MatMul
|
||||
mindspore.ops.MatrixInverse
|
||||
mindspore.ops.Svd
|
||||
mindspore.ops.TridiagonalMatMul
|
||||
|
||||
Tensor操作算子
|
||||
----------------
|
||||
|
|
@ -415,6 +424,7 @@ Tensor创建
|
|||
mindspore.ops.Randperm
|
||||
mindspore.ops.StandardLaplace
|
||||
mindspore.ops.StandardNormal
|
||||
mindspore.ops.TruncatedNormal
|
||||
mindspore.ops.UniformInt
|
||||
mindspore.ops.UniformReal
|
||||
|
||||
|
|
@ -476,9 +486,22 @@ Array操作
|
|||
mindspore.ops.Range
|
||||
mindspore.ops.Rank
|
||||
mindspore.ops.Reshape
|
||||
mindspore.ops.ResizeNearestNeighborV2
|
||||
mindspore.ops.ReverseSequence
|
||||
mindspore.ops.ReverseV2
|
||||
mindspore.ops.RightShift
|
||||
mindspore.ops.ScatterAddWithAxis
|
||||
mindspore.ops.ScatterNd
|
||||
mindspore.ops.ScatterNdDiv
|
||||
mindspore.ops.ScatterNdMax
|
||||
mindspore.ops.ScatterNdMin
|
||||
mindspore.ops.ScatterNdMul
|
||||
mindspore.ops.SearchSorted
|
||||
mindspore.ops.SegmentMax
|
||||
mindspore.ops.SegmentMean
|
||||
mindspore.ops.SegmentMin
|
||||
mindspore.ops.SegmentProd
|
||||
mindspore.ops.SegmentSum
|
||||
mindspore.ops.Select
|
||||
mindspore.ops.Shape
|
||||
mindspore.ops.Size
|
||||
|
|
@ -488,8 +511,10 @@ Array操作
|
|||
mindspore.ops.SpaceToDepth
|
||||
mindspore.ops.SparseGatherV2
|
||||
mindspore.ops.Split
|
||||
mindspore.ops.SplitV
|
||||
mindspore.ops.Squeeze
|
||||
mindspore.ops.Stack
|
||||
mindspore.ops.STFT
|
||||
mindspore.ops.StridedSlice
|
||||
mindspore.ops.TensorScatterAdd
|
||||
mindspore.ops.TensorScatterDiv
|
||||
|
|
@ -500,16 +525,22 @@ Array操作
|
|||
mindspore.ops.TensorScatterUpdate
|
||||
mindspore.ops.TensorShape
|
||||
mindspore.ops.Tile
|
||||
mindspore.ops.Trace
|
||||
mindspore.ops.Transpose
|
||||
mindspore.ops.Tril
|
||||
mindspore.ops.TrilIndices
|
||||
mindspore.ops.Triu
|
||||
mindspore.ops.TriuIndices
|
||||
mindspore.ops.Unique
|
||||
mindspore.ops.UniqueConsecutive
|
||||
mindspore.ops.UniqueWithPad
|
||||
mindspore.ops.UnravelIndex
|
||||
mindspore.ops.UnsortedSegmentMax
|
||||
mindspore.ops.UnsortedSegmentMin
|
||||
mindspore.ops.UnsortedSegmentProd
|
||||
mindspore.ops.UnsortedSegmentSum
|
||||
mindspore.ops.Unstack
|
||||
mindspore.ops.SplitV
|
||||
mindspore.ops.STFT
|
||||
mindspore.ops.UpperBound
|
||||
|
||||
类型转换
|
||||
^^^^^^^^^^^^^^^^
|
||||
|
|
|
|||
|
|
@ -74,6 +74,8 @@ Neural Network
|
|||
mindspore.ops.Padding
|
||||
mindspore.ops.ResizeNearestNeighbor
|
||||
mindspore.ops.ResizeBilinear
|
||||
mindspore.ops.UpsampleNearest3D
|
||||
mindspore.ops.UpsampleTrilinear3D
|
||||
|
||||
Loss Function
|
||||
^^^^^^^^^^^^^
|
||||
|
|
@ -96,6 +98,7 @@ Loss Function
|
|||
mindspore.ops.SoftMarginLoss
|
||||
mindspore.ops.SoftmaxCrossEntropyWithLogits
|
||||
mindspore.ops.SparseSoftmaxCrossEntropyWithLogits
|
||||
mindspore.ops.TripletMarginLoss
|
||||
|
||||
Activation Function
|
||||
^^^^^^^^^^^^^^^^^^^
|
||||
|
|
@ -203,7 +206,10 @@ Image Processing
|
|||
mindspore.ops.IOU
|
||||
mindspore.ops.L2Normalize
|
||||
mindspore.ops.NMSWithMask
|
||||
mindspore.ops.RGBToHSV
|
||||
mindspore.ops.ROIAlign
|
||||
mindspore.ops.SampleDistortedBoundingBoxV2
|
||||
mindspore.ops.ScaleAndTranslate
|
||||
|
||||
Text Processing
|
||||
^^^^^^^^^^^^^^^
|
||||
|
|
@ -318,6 +324,7 @@ Element-wise Operator
|
|||
mindspore.ops.TruncateMod
|
||||
mindspore.ops.Xdivy
|
||||
mindspore.ops.Xlogy
|
||||
mindspore.ops.Zeta
|
||||
|
||||
|
||||
Reduction Operator
|
||||
|
|
@ -378,6 +385,7 @@ Linear Algebraic Operator
|
|||
mindspore.ops.MatMul
|
||||
mindspore.ops.MatrixInverse
|
||||
mindspore.ops.Svd
|
||||
mindspore.ops.TridiagonalMatMul
|
||||
|
||||
Tensor Operation Operator
|
||||
--------------------------
|
||||
|
|
@ -415,6 +423,7 @@ Random Generation Operator
|
|||
mindspore.ops.Randperm
|
||||
mindspore.ops.StandardLaplace
|
||||
mindspore.ops.StandardNormal
|
||||
mindspore.ops.TruncatedNormal
|
||||
mindspore.ops.UniformInt
|
||||
mindspore.ops.UniformReal
|
||||
|
||||
|
|
@ -476,9 +485,22 @@ Array Operation
|
|||
mindspore.ops.Range
|
||||
mindspore.ops.Rank
|
||||
mindspore.ops.Reshape
|
||||
mindspore.ops.ResizeNearestNeighborV2
|
||||
mindspore.ops.ReverseSequence
|
||||
mindspore.ops.ReverseV2
|
||||
mindspore.ops.RightShift
|
||||
mindspore.ops.ScatterAddWithAxis
|
||||
mindspore.ops.ScatterNd
|
||||
mindspore.ops.ScatterNdDiv
|
||||
mindspore.ops.ScatterNdMax
|
||||
mindspore.ops.ScatterNdMin
|
||||
mindspore.ops.ScatterNdMul
|
||||
mindspore.ops.SearchSorted
|
||||
mindspore.ops.SegmentMax
|
||||
mindspore.ops.SegmentMean
|
||||
mindspore.ops.SegmentMin
|
||||
mindspore.ops.SegmentProd
|
||||
mindspore.ops.SegmentSum
|
||||
mindspore.ops.Select
|
||||
mindspore.ops.Shape
|
||||
mindspore.ops.Size
|
||||
|
|
@ -488,8 +510,10 @@ Array Operation
|
|||
mindspore.ops.SpaceToDepth
|
||||
mindspore.ops.SparseGatherV2
|
||||
mindspore.ops.Split
|
||||
mindspore.ops.SplitV
|
||||
mindspore.ops.Squeeze
|
||||
mindspore.ops.Stack
|
||||
mindspore.ops.STFT
|
||||
mindspore.ops.StridedSlice
|
||||
mindspore.ops.TensorScatterAdd
|
||||
mindspore.ops.TensorScatterDiv
|
||||
|
|
@ -500,16 +524,22 @@ Array Operation
|
|||
mindspore.ops.TensorScatterUpdate
|
||||
mindspore.ops.TensorShape
|
||||
mindspore.ops.Tile
|
||||
mindspore.ops.Trace
|
||||
mindspore.ops.Transpose
|
||||
mindspore.ops.Tril
|
||||
mindspore.ops.TrilIndices
|
||||
mindspore.ops.Triu
|
||||
mindspore.ops.TriuIndices
|
||||
mindspore.ops.Unique
|
||||
mindspore.ops.UniqueConsecutive
|
||||
mindspore.ops.UniqueWithPad
|
||||
mindspore.ops.UnravelIndex
|
||||
mindspore.ops.UnsortedSegmentMax
|
||||
mindspore.ops.UnsortedSegmentMin
|
||||
mindspore.ops.UnsortedSegmentProd
|
||||
mindspore.ops.UnsortedSegmentSum
|
||||
mindspore.ops.Unstack
|
||||
mindspore.ops.SplitV
|
||||
mindspore.ops.STFT
|
||||
mindspore.ops.UpperBound
|
||||
|
||||
Type Conversion
|
||||
^^^^^^^^^^^^^^^
|
||||
|
|
|
|||
|
|
@ -46,7 +46,9 @@ from .array_ops import (ArgMaxWithValue, ArgMinWithValue, Argmax, Argmin, BatchT
|
|||
UnsortedSegmentSum, Unstack, UpperBound, Zeros, ZerosLike, AffineGrid, Bincount, CheckNumerics,
|
||||
HammingWindow, IdentityN, IndexFill, LeftShift, ListDiff, LogSpace, MatrixBandPart,
|
||||
MatrixDiagPartV3, MatrixDiagV3, MatrixSetDiagV3, NonZero, Expand, Col2Im, ConjugateTranspose,
|
||||
FillDiagonal, Fills)
|
||||
FillDiagonal, Fills, ResizeNearestNeighborV2, RightShift, ScatterAddWithAxis,
|
||||
ScatterNdMul, SegmentMean, SegmentProd, SegmentSum, SegmentMax, SegmentMin, Tril, Triu,
|
||||
UniqueConsecutive, UnravelIndex)
|
||||
from .comm_ops import (AllGather, AllReduce, NeighborExchange, NeighborExchangeV2, AlltoAll, _AllSwap, ReduceScatter,
|
||||
Broadcast,
|
||||
_MirrorOperator, _MirrorMiniStepOperator, _MiniStepAllGather, ReduceOp, _VirtualDataset,
|
||||
|
|
@ -59,7 +61,7 @@ from .debug_ops import (ImageSummary, InsertGradientOf, HookBackward, ScalarSumm
|
|||
TensorSummary, HistogramSummary, Print, Assert)
|
||||
from .image_ops import (CropAndResize, NonMaxSuppressionV3, HSVToRGB, AdjustHue, AdjustSaturation,
|
||||
NonMaxSuppressionWithOverlaps, ResizeArea, ResizeBicubic, ResizeBilinearV2, ExtractGlimpse,
|
||||
CombinedNonMaxSuppression)
|
||||
CombinedNonMaxSuppression, RGBToHSV, ScaleAndTranslate)
|
||||
from .inner_ops import (ScalarCast, Randperm, NoRepeatNGram, LambApplyOptimizerAssign, LambApplyWeightAssign,
|
||||
FusedWeightScaleApplyMomentum, FusedCastAdamWeightDecay, FusedAdaFactor,
|
||||
FusedAdaFactorWithGlobalNorm)
|
||||
|
|
@ -83,7 +85,7 @@ from .math_ops import (Abs, ACos, Asin, Asinh, AddN, AccumulateNV2, AssignAdd, A
|
|||
FFTWithSize, Heaviside, Histogram, Hypot, Lcm, LuUnpack, MatrixExp,
|
||||
MatrixLogarithm, MatrixPower, MatrixSolve, MatrixTriangularSolve, ReduceStd, STFT,
|
||||
NextAfter, Orgqr, Qr, RaggedRange, Digamma, Eig, EuclideanNorm, CompareAndBitpack, ComplexAbs,
|
||||
CumulativeLogsumexp, Gcd)
|
||||
CumulativeLogsumexp, Gcd, Trace, TridiagonalMatMul, TrilIndices, TriuIndices, Zeta)
|
||||
from .nn_ops import (LSTM, SGD, Adam, AdamWeightDecay, FusedSparseAdam, FusedSparseLazyAdam, AdamNoUpdateParam,
|
||||
ApplyMomentum, BatchNorm, BiasAdd, Conv2D, Conv3D, Conv2DTranspose, Conv3DTranspose,
|
||||
DepthwiseConv2dNative,
|
||||
|
|
@ -110,10 +112,11 @@ from .nn_ops import (LSTM, SGD, Adam, AdamWeightDecay, FusedSparseAdam, FusedSpa
|
|||
GridSampler3D, MaxPool3DWithArgmax, MaxUnpool2D, NuclearNorm, NthElement, MultilabelMarginLoss,
|
||||
PSROIPooling, Dilation2D, DataFormatVecPermute, DeformableOffsets, FractionalAvgPool,
|
||||
FractionalMaxPool, FractionalMaxPool3DWithFixedKsize, FractionalMaxPoolWithFixedKsize,
|
||||
GridSampler2D)
|
||||
GridSampler2D, TripletMarginLoss, UpsampleNearest3D, UpsampleTrilinear3D)
|
||||
from .other_ops import (Assign, IOU, BartlettWindow, BlackmanWindow, BoundingBoxDecode, BoundingBoxEncode,
|
||||
ConfusionMatrix, UpdateState, Load, StopGradient,
|
||||
CheckValid, Partial, Depend, identity, Push, Pull, PyFunc, _DynamicLossScale)
|
||||
CheckValid, Partial, Depend, identity, Push, Pull, PyFunc, _DynamicLossScale,
|
||||
SampleDistortedBoundingBoxV2)
|
||||
from .random_ops import (RandomChoiceWithMask, StandardNormal, Gamma, RandomGamma, Poisson, UniformInt, UniformReal,
|
||||
RandomCategorical, StandardLaplace, Multinomial, UniformCandidateSampler,
|
||||
LogUniformCandidateSampler, TruncatedNormal, LogNormalReverse, NonDeterministicInts,
|
||||
|
|
@ -586,7 +589,6 @@ __all__ = [
|
|||
"ParameterizedTruncatedNormal",
|
||||
"RandomGamma",
|
||||
"RandomPoisson",
|
||||
"PadAndShift",
|
||||
"RaggedRange",
|
||||
"Digamma",
|
||||
"Dilation2D",
|
||||
|
|
@ -614,6 +616,30 @@ __all__ = [
|
|||
"FractionalMaxPool",
|
||||
"FractionalMaxPool3DWithFixedKsize",
|
||||
"FractionalMaxPoolWithFixedKsize",
|
||||
"RGBToHSV",
|
||||
"ResizeNearestNeighborV2",
|
||||
"RightShift",
|
||||
"SampleDistortedBoundingBoxV2",
|
||||
"ScaleAndTranslate",
|
||||
"ScatterAddWithAxis",
|
||||
"ScatterNdMul",
|
||||
"SegmentMean",
|
||||
"SegmentProd",
|
||||
"SegmentSum",
|
||||
"SegmentMax",
|
||||
"SegmentMin",
|
||||
"TripletMarginLoss",
|
||||
"UpsampleNearest3D",
|
||||
"UpsampleTrilinear3D",
|
||||
"Trace",
|
||||
"TridiagonalMatMul",
|
||||
"Tril",
|
||||
"TrilIndices",
|
||||
"Triu",
|
||||
"TriuIndices",
|
||||
"UniqueConsecutive",
|
||||
"UnravelIndex",
|
||||
"Zeta",
|
||||
]
|
||||
|
||||
__custom__ = [
|
||||
|
|
|
|||
Loading…
Reference in New Issue