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.

operations.rst 12 kB

4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432
  1. operations
  2. ----------
  3. 神经网络算子
  4. ^^^^^^^^^^^^^^^^^^^^^^^^
  5. operations中的Primitive算子在使用前需要实例化。
  6. .. mscnplatformautosummary::
  7. :toctree: ops
  8. :nosignatures:
  9. :template: classtemplate.rst
  10. mindspore.ops.Acosh
  11. mindspore.ops.Adam
  12. mindspore.ops.AdamNoUpdateParam
  13. mindspore.ops.AdamWeightDecay
  14. mindspore.ops.AdaptiveAvgPool2D
  15. mindspore.ops.ApplyAdadelta
  16. mindspore.ops.ApplyAdagrad
  17. mindspore.ops.ApplyAdagradDA
  18. mindspore.ops.ApplyAdagradV2
  19. mindspore.ops.ApplyAdaMax
  20. mindspore.ops.ApplyAddSign
  21. mindspore.ops.ApplyCenteredRMSProp
  22. mindspore.ops.ApplyGradientDescent
  23. mindspore.ops.ApplyMomentum
  24. mindspore.ops.ApplyPowerSign
  25. mindspore.ops.ApplyProximalAdagrad
  26. mindspore.ops.ApplyProximalGradientDescent
  27. mindspore.ops.ApplyRMSProp
  28. mindspore.ops.AvgPool
  29. mindspore.ops.AvgPool3D
  30. mindspore.ops.BasicLSTMCell
  31. mindspore.ops.BatchNorm
  32. mindspore.ops.BCEWithLogitsLoss
  33. mindspore.ops.BiasAdd
  34. mindspore.ops.BinaryCrossEntropy
  35. mindspore.ops.ComputeAccidentalHits
  36. mindspore.ops.Conv2D
  37. mindspore.ops.Conv2DBackpropInput
  38. mindspore.ops.Conv2DTranspose
  39. mindspore.ops.Conv3D
  40. mindspore.ops.Conv3DTranspose
  41. mindspore.ops.CTCGreedyDecoder
  42. mindspore.ops.CTCLoss
  43. mindspore.ops.DataFormatDimMap
  44. mindspore.ops.DepthwiseConv2dNative
  45. mindspore.ops.Dropout
  46. mindspore.ops.Dropout2D
  47. mindspore.ops.Dropout3D
  48. mindspore.ops.DropoutDoMask
  49. mindspore.ops.DropoutGenMask
  50. mindspore.ops.DynamicGRUV2
  51. mindspore.ops.DynamicRNN
  52. mindspore.ops.Elu
  53. mindspore.ops.FastGeLU
  54. mindspore.ops.Flatten
  55. mindspore.ops.FloorMod
  56. mindspore.ops.FusedSparseAdam
  57. mindspore.ops.FusedSparseLazyAdam
  58. mindspore.ops.FusedSparseProximalAdagrad
  59. mindspore.ops.GeLU
  60. mindspore.ops.GetNext
  61. mindspore.ops.HShrink
  62. mindspore.ops.HSigmoid
  63. mindspore.ops.HSwish
  64. mindspore.ops.KLDivLoss
  65. mindspore.ops.L2Loss
  66. mindspore.ops.L2Normalize
  67. mindspore.ops.LARSUpdate
  68. mindspore.ops.LayerNorm
  69. mindspore.ops.LogSoftmax
  70. mindspore.ops.LRN
  71. mindspore.ops.LSTM
  72. mindspore.ops.MaxPool
  73. mindspore.ops.MaxPool3D
  74. mindspore.ops.MaxPoolWithArgmax
  75. mindspore.ops.MirrorPad
  76. mindspore.ops.Mish
  77. mindspore.ops.NLLLoss
  78. mindspore.ops.OneHot
  79. mindspore.ops.Pad
  80. mindspore.ops.PReLU
  81. mindspore.ops.ReLU
  82. mindspore.ops.ReLU6
  83. mindspore.ops.ReLUV2
  84. mindspore.ops.ResizeBilinear
  85. mindspore.ops.RNNTLoss
  86. mindspore.ops.ROIAlign
  87. mindspore.ops.SeLU
  88. mindspore.ops.SGD
  89. mindspore.ops.Sigmoid
  90. mindspore.ops.SigmoidCrossEntropyWithLogits
  91. mindspore.ops.SmoothL1Loss
  92. mindspore.ops.SoftMarginLoss
  93. mindspore.ops.Softmax
  94. mindspore.ops.SoftmaxCrossEntropyWithLogits
  95. mindspore.ops.Softplus
  96. mindspore.ops.SoftShrink
  97. mindspore.ops.Softsign
  98. mindspore.ops.SparseApplyAdagrad
  99. mindspore.ops.SparseApplyAdagradV2
  100. mindspore.ops.SparseApplyProximalAdagrad
  101. mindspore.ops.SparseSoftmaxCrossEntropyWithLogits
  102. mindspore.ops.Stack
  103. mindspore.ops.Tanh
  104. mindspore.ops.TopK
  105. mindspore.ops.Unstack
  106. 数学算子
  107. ^^^^^^^^^^^^^^^
  108. .. mscnplatformautosummary::
  109. :toctree: ops
  110. :nosignatures:
  111. :template: classtemplate.rst
  112. mindspore.ops.Abs
  113. mindspore.ops.AccumulateNV2
  114. mindspore.ops.ACos
  115. mindspore.ops.Add
  116. mindspore.ops.AddN
  117. mindspore.ops.ApproximateEqual
  118. mindspore.ops.Asin
  119. mindspore.ops.Asinh
  120. mindspore.ops.AssignAdd
  121. mindspore.ops.AssignSub
  122. mindspore.ops.Atan
  123. mindspore.ops.Atan2
  124. mindspore.ops.Atanh
  125. mindspore.ops.BatchMatMul
  126. mindspore.ops.BesselI0e
  127. mindspore.ops.BesselI1e
  128. mindspore.ops.BitwiseAnd
  129. mindspore.ops.BitwiseOr
  130. mindspore.ops.BitwiseXor
  131. mindspore.ops.Cdist
  132. mindspore.ops.Ceil
  133. mindspore.ops.Conj
  134. mindspore.ops.Cos
  135. mindspore.ops.Cosh
  136. mindspore.ops.CumProd
  137. mindspore.ops.CumSum
  138. mindspore.ops.Div
  139. mindspore.ops.DivNoNan
  140. mindspore.ops.Eps
  141. mindspore.ops.Equal
  142. mindspore.ops.EqualCount
  143. mindspore.ops.Erf
  144. mindspore.ops.Erfc
  145. mindspore.ops.Erfinv
  146. mindspore.ops.Exp
  147. mindspore.ops.Expm1
  148. mindspore.ops.FloatStatus
  149. mindspore.ops.Floor
  150. mindspore.ops.FloorDiv
  151. mindspore.ops.Ger
  152. mindspore.ops.Greater
  153. mindspore.ops.GreaterEqual
  154. mindspore.ops.HistogramFixedWidth
  155. mindspore.ops.Imag
  156. mindspore.ops.IndexAdd
  157. mindspore.ops.InplaceAdd
  158. mindspore.ops.InplaceSub
  159. mindspore.ops.Inv
  160. mindspore.ops.Invert
  161. mindspore.ops.IsInf
  162. mindspore.ops.IsNan
  163. mindspore.ops.Lerp
  164. mindspore.ops.Less
  165. mindspore.ops.LessEqual
  166. mindspore.ops.LinSpace
  167. mindspore.ops.Log
  168. mindspore.ops.Log1p
  169. mindspore.ops.LogicalAnd
  170. mindspore.ops.LogicalNot
  171. mindspore.ops.LogicalOr
  172. mindspore.ops.LpNorm
  173. mindspore.ops.MatMul
  174. mindspore.ops.MatrixInverse
  175. mindspore.ops.Maximum
  176. mindspore.ops.Minimum
  177. mindspore.ops.Mod
  178. mindspore.ops.Mul
  179. mindspore.ops.MulNoNan
  180. mindspore.ops.Neg
  181. mindspore.ops.NMSWithMask
  182. mindspore.ops.NotEqual
  183. mindspore.ops.NPUAllocFloatStatus
  184. mindspore.ops.NPUClearFloatStatus
  185. mindspore.ops.NPUGetFloatStatus
  186. mindspore.ops.Pow
  187. mindspore.ops.Real
  188. mindspore.ops.RealDiv
  189. mindspore.ops.Reciprocal
  190. mindspore.ops.ReduceAll
  191. mindspore.ops.ReduceAny
  192. mindspore.ops.ReduceMax
  193. mindspore.ops.ReduceMean
  194. mindspore.ops.ReduceMin
  195. mindspore.ops.ReduceProd
  196. mindspore.ops.ReduceSum
  197. mindspore.ops.Round
  198. mindspore.ops.Rsqrt
  199. mindspore.ops.Sign
  200. mindspore.ops.Sin
  201. mindspore.ops.Sinh
  202. mindspore.ops.Sqrt
  203. mindspore.ops.Square
  204. mindspore.ops.SquaredDifference
  205. mindspore.ops.SquareSumAll
  206. mindspore.ops.Sub
  207. mindspore.ops.Tan
  208. mindspore.ops.TruncateDiv
  209. mindspore.ops.TruncateMod
  210. mindspore.ops.Xdivy
  211. mindspore.ops.Xlogy
  212. 数组算子
  213. ^^^^^^^^^^^^^^^^
  214. .. mscnplatformautosummary::
  215. :toctree: ops
  216. :nosignatures:
  217. :template: classtemplate.rst
  218. mindspore.ops.ApplyFtrl
  219. mindspore.ops.Argmax
  220. mindspore.ops.ArgMaxWithValue
  221. mindspore.ops.Argmin
  222. mindspore.ops.ArgMinWithValue
  223. mindspore.ops.BatchToSpace
  224. mindspore.ops.BatchToSpaceND
  225. mindspore.ops.BroadcastTo
  226. mindspore.ops.Cast
  227. mindspore.ops.Concat
  228. mindspore.ops.DepthToSpace
  229. mindspore.ops.DType
  230. mindspore.ops.DynamicShape
  231. mindspore.ops.EditDistance
  232. mindspore.ops.EmbeddingLookup
  233. mindspore.ops.ExpandDims
  234. mindspore.ops.ExtractVolumePatches
  235. mindspore.ops.Eye
  236. mindspore.ops.Fill
  237. mindspore.ops.FusedSparseFtrl
  238. mindspore.ops.Gather
  239. mindspore.ops.GatherD
  240. mindspore.ops.GatherNd
  241. mindspore.ops.Identity
  242. mindspore.ops.InplaceUpdate
  243. mindspore.ops.InvertPermutation
  244. mindspore.ops.IsFinite
  245. mindspore.ops.IsInstance
  246. mindspore.ops.IsSubClass
  247. mindspore.ops.MaskedFill
  248. mindspore.ops.MaskedSelect
  249. mindspore.ops.Meshgrid
  250. mindspore.ops.Ones
  251. mindspore.ops.OnesLike
  252. mindspore.ops.Padding
  253. mindspore.ops.ParallelConcat
  254. mindspore.ops.Randperm
  255. mindspore.ops.Rank
  256. mindspore.ops.Reshape
  257. mindspore.ops.ResizeNearestNeighbor
  258. mindspore.ops.ReverseSequence
  259. mindspore.ops.ReverseV2
  260. mindspore.ops.Rint
  261. mindspore.ops.SameTypeShape
  262. mindspore.ops.ScalarCast
  263. mindspore.ops.ScalarToArray
  264. mindspore.ops.ScalarToTensor
  265. mindspore.ops.ScatterAdd
  266. mindspore.ops.ScatterDiv
  267. mindspore.ops.ScatterMax
  268. mindspore.ops.ScatterMin
  269. mindspore.ops.ScatterMul
  270. mindspore.ops.ScatterNd
  271. mindspore.ops.ScatterNdAdd
  272. mindspore.ops.ScatterNdSub
  273. mindspore.ops.ScatterNdUpdate
  274. mindspore.ops.ScatterNonAliasingAdd
  275. mindspore.ops.ScatterSub
  276. mindspore.ops.ScatterUpdate
  277. mindspore.ops.Select
  278. mindspore.ops.Shape
  279. mindspore.ops.Size
  280. mindspore.ops.Slice
  281. mindspore.ops.Sort
  282. mindspore.ops.SpaceToBatch
  283. mindspore.ops.SpaceToBatchND
  284. mindspore.ops.SpaceToDepth
  285. mindspore.ops.SparseApplyFtrl
  286. mindspore.ops.SparseApplyFtrlV2
  287. mindspore.ops.SparseGatherV2
  288. mindspore.ops.Split
  289. mindspore.ops.SplitV
  290. mindspore.ops.Squeeze
  291. mindspore.ops.StridedSlice
  292. mindspore.ops.TensorScatterAdd
  293. mindspore.ops.TensorScatterMax
  294. mindspore.ops.TensorScatterMin
  295. mindspore.ops.TensorScatterSub
  296. mindspore.ops.TensorScatterUpdate
  297. mindspore.ops.TensorShape
  298. mindspore.ops.Tile
  299. mindspore.ops.Transpose
  300. mindspore.ops.TupleToArray
  301. mindspore.ops.Unique
  302. mindspore.ops.UniqueWithPad
  303. mindspore.ops.UnsortedSegmentMax
  304. mindspore.ops.UnsortedSegmentMin
  305. mindspore.ops.UnsortedSegmentProd
  306. mindspore.ops.UnsortedSegmentSum
  307. mindspore.ops.Zeros
  308. mindspore.ops.ZerosLike
  309. 通信算子
  310. ^^^^^^^^^^^^^^^^^^^^^^^
  311. 注意,以下列表中的接口需要先配置好通信环境变量。
  312. 针对Ascend设备,用户需要准备rank表,设置rank_id和device_id,详见 `Ascend指导文档 \
  313. <https://www.mindspore.cn/tutorials/zh-CN/master/intermediate/distributed_training/
  314. distributed_training_ascend.html#id12>`_ 。
  315. 针对GPU设备,用户需要准备host文件和mpi,详见 `GPU指导文档 \
  316. <https://www.mindspore.cn/tutorials/zh-CN/master/intermediate/distributed_training/distributed_training_gpu.html#id6>`_ 。
  317. .. mscnplatformautosummary::
  318. :toctree: ops
  319. :nosignatures:
  320. :template: classtemplate.rst
  321. mindspore.ops.AllGather
  322. mindspore.ops.AllReduce
  323. mindspore.ops.AlltoAll
  324. mindspore.ops.Broadcast
  325. mindspore.ops.NeighborExchange
  326. mindspore.ops.NeighborExchangeV2
  327. mindspore.ops.ReduceOp
  328. mindspore.ops.ReduceScatter
  329. 调试算子
  330. ^^^^^^^^^^^^^^^
  331. .. mscnplatformautosummary::
  332. :toctree: ops
  333. :nosignatures:
  334. :template: classtemplate.rst
  335. mindspore.ops.HistogramSummary
  336. mindspore.ops.HookBackward
  337. mindspore.ops.ImageSummary
  338. mindspore.ops.InsertGradientOf
  339. mindspore.ops.Print
  340. mindspore.ops.ScalarSummary
  341. mindspore.ops.TensorSummary
  342. 随机算子
  343. ^^^^^^^^^^^^^^^^^
  344. .. mscnplatformautosummary::
  345. :toctree: ops
  346. :nosignatures:
  347. :template: classtemplate.rst
  348. mindspore.ops.Gamma
  349. mindspore.ops.LogUniformCandidateSampler
  350. mindspore.ops.Multinomial
  351. mindspore.ops.Poisson
  352. mindspore.ops.RandomCategorical
  353. mindspore.ops.RandomChoiceWithMask
  354. mindspore.ops.StandardLaplace
  355. mindspore.ops.StandardNormal
  356. mindspore.ops.UniformCandidateSampler
  357. mindspore.ops.UniformInt
  358. mindspore.ops.UniformReal
  359. 图像算子
  360. ^^^^^^^^^^^^^^^
  361. .. mscnplatformautosummary::
  362. :toctree: ops
  363. :nosignatures:
  364. :template: classtemplate.rst
  365. mindspore.ops.CropAndResize
  366. 稀疏算子
  367. ^^^^^^^^^^^^^^^^
  368. .. mscnplatformautosummary::
  369. :toctree: ops
  370. :nosignatures:
  371. :template: classtemplate.rst
  372. mindspore.ops.SparseToDense
  373. mindspore.ops.SparseTensorDenseMatmul
  374. 自定义算子
  375. ^^^^^^^^^^^^^^^^
  376. .. mscnplatformautosummary::
  377. :toctree: ops
  378. :nosignatures:
  379. :template: classtemplate.rst
  380. mindspore.ops.Custom
  381. 其他算子
  382. ^^^^^^^^^^^^^^^
  383. .. mscnplatformautosummary::
  384. :toctree: ops
  385. :nosignatures:
  386. :template: classtemplate.rst
  387. mindspore.ops.Assign
  388. mindspore.ops.BoundingBoxDecode
  389. mindspore.ops.BoundingBoxEncode
  390. mindspore.ops.CheckBprop
  391. mindspore.ops.CheckValid
  392. mindspore.ops.Depend
  393. mindspore.ops.InTopK
  394. mindspore.ops.IOU
  395. mindspore.ops.NoRepeatNGram
  396. mindspore.ops.Partial
  397. mindspore.ops.PopulationCount