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.

mindspore.ops.rst 15 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
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
4 years ago
4 years ago
4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401
  1. mindspore.ops
  2. =============
  3. 可用于Cell的构造函数的算子。
  4. .. code-block::
  5. import mindspore.ops as ops
  6. MindSpore中 `mindspore.ops` 接口与上一版本相比,新增、删除和支持平台的变化信息请参考 `API Updates <https://gitee.com/mindspore/docs/blob/master/resource/api_updates/ops_api_updates.md>`_ 。
  7. .. include:: operations.rst
  8. functional
  9. -----------
  10. functional算子是经过初始化后的Primitive,可以直接作为函数使用。functional算子的使用示例如下:
  11. .. code-block:: python
  12. from mindspore import Tensor, ops
  13. from mindspore import dtype as mstype
  14. input_x = Tensor(-1, mstype.int32)
  15. input_dict = {'x':1, 'y':2}
  16. result_abs = ops.absolute(input_x)
  17. print(result_abs)
  18. result_in_dict = ops.in_dict('x', input_dict)
  19. print(result_in_dict)
  20. result_not_in_dict = ops.not_in_dict('x', input_dict)
  21. print(result_not_in_dict)
  22. result_isconstant = ops.isconstant(input_x)
  23. print(result_isconstant)
  24. result_typeof = ops.typeof(input_x)
  25. print(result_typeof)
  26. # outputs:
  27. # 1
  28. # True
  29. # False
  30. # True
  31. # Tensor[Int32]
  32. .. list-table::
  33. :widths: 50 50
  34. :header-rows: 1
  35. * - functional
  36. - Description
  37. * - mindspore.ops.absolute
  38. - Refer to :class:`mindspore.ops.Abs`.
  39. * - mindspore.ops.acos
  40. - Refer to :class:`mindspore.ops.ACos`.
  41. * - mindspore.ops.acosh
  42. - Refer to :class:`mindspore.ops.Acosh`.
  43. * - mindspore.ops.add
  44. - Refer to :class:`mindspore.ops.Add`.
  45. * - mindspore.ops.addn
  46. - Refer to :class:`mindspore.ops.AddN`.
  47. * - mindspore.ops.asin
  48. - Refer to :class:`mindspore.ops.Asin`.
  49. * - mindspore.ops.asinh
  50. - Refer to :class:`mindspore.ops.Asinh`.
  51. * - mindspore.ops.assign
  52. - Refer to :class:`mindspore.ops.Assign`.
  53. * - mindspore.ops.assign_add
  54. - Refer to :class:`mindspore.ops.AssignAdd`.
  55. * - mindspore.ops.assign_sub
  56. - Refer to :class:`mindspore.ops.AssignSub`.
  57. * - mindspore.ops.atan
  58. - Refer to :class:`mindspore.ops.Atan`.
  59. * - mindspore.ops.atan2
  60. - Refer to :class:`mindspore.ops.Atan2`.
  61. * - mindspore.ops.atanh
  62. - Refer to :class:`mindspore.ops.Atanh`.
  63. * - mindspore.ops.bitwise_and
  64. - Refer to :class:`mindspore.ops.BitwiseAnd`.
  65. * - mindspore.ops.bitwise_or
  66. - Refer to :class:`mindspore.ops.BitwiseOr`.
  67. * - mindspore.ops.bitwise_xor
  68. - Refer to :class:`mindspore.ops.BitwiseXor`.
  69. * - mindspore.ops.bool_and
  70. - Calculate the result of logical AND operation. (Usage is the same as "and" in Python)
  71. * - mindspore.ops.bool_eq
  72. - Determine whether the Boolean values are equal. (Usage is the same as "==" in Python)
  73. * - mindspore.ops.bool_not
  74. - Calculate the result of logical NOT operation. (Usage is the same as "not" in Python)
  75. * - mindspore.ops.bool_or
  76. - Calculate the result of logical OR operation. (Usage is the same as "or" in Python)
  77. * - mindspore.ops.cast
  78. - Refer to :class:`mindspore.ops.Cast`.
  79. * - mindspore.ops.check_bprop
  80. - Refer to :class:`mindspore.ops.CheckBprop`.
  81. * - mindspore.ops.cos
  82. - Refer to :class:`mindspore.ops.Cos`.
  83. * - mindspore.ops.cosh
  84. - Refer to :class:`mindspore.ops.Cosh`.
  85. * - mindspore.ops.cumprod
  86. - Refer to :class:`mindspore.ops.CumProd`.
  87. * - mindspore.ops.cumsum
  88. - Refer to :class:`mindspore.ops.CumSum`.
  89. * - mindspore.ops.div
  90. - Refer to :class:`mindspore.ops.RealDiv`.
  91. * - mindspore.ops.depend
  92. - Refer to :class:`mindspore.ops.Depend`.
  93. * - mindspore.ops.dtype
  94. - Refer to :class:`mindspore.ops.DType`.
  95. * - mindspore.ops.erf
  96. - Refer to :class:`mindspore.ops.Erf`.
  97. * - mindspore.ops.erfc
  98. - Refer to :class:`mindspore.ops.Erfc`.
  99. * - mindspore.ops.eye
  100. - Refer to :class:`mindspore.ops.Eye`.
  101. * - mindspore.ops.equal
  102. - Refer to :class:`mindspore.ops.Equal`.
  103. * - mindspore.ops.expand_dims
  104. - Refer to :class:`mindspore.ops.ExpandDims`.
  105. * - mindspore.ops.exp
  106. - Refer to :class:`mindspore.ops.Exp`.
  107. * - mindspore.ops.fill
  108. - Refer to :class:`mindspore.ops.Fill`.
  109. * - mindspore.ops.floor
  110. - Refer to :class:`mindspore.ops.Floor`.
  111. * - mindspore.ops.floordiv
  112. - Refer to :class:`mindspore.ops.FloorDiv`.
  113. * - mindspore.ops.floormod
  114. - Refer to :class:`mindspore.ops.FloorMod`.
  115. * - mindspore.ops.gather
  116. - Refer to :class:`mindspore.ops.Gather`.
  117. * - mindspore.ops.gather_d
  118. - Refer to :class:`mindspore.ops.GatherD`.
  119. * - mindspore.ops.gather_nd
  120. - Refer to :class:`mindspore.ops.GatherNd`.
  121. * - mindspore.ops.ge
  122. - Refer to :class:`mindspore.ops.GreaterEqual`.
  123. * - mindspore.ops.gt
  124. - Refer to :class:`mindspore.ops.Greater`.
  125. * - mindspore.ops.invert
  126. - Refer to :class:`mindspore.ops.Invert`.
  127. * - mindspore.ops.in_dict
  128. - Determine if a str in dict.
  129. * - mindspore.ops.is_not
  130. - Determine whether the input is not the same as the other one. (Usage is the same as "is not" in Python)
  131. * - mindspore.ops.is\_
  132. - Determine whether the input is the same as the other one. (Usage is the same as "is" in Python)
  133. * - mindspore.ops.isconstant
  134. - Determine whether the object is constant.
  135. * - mindspore.ops.isfinite
  136. - Refer to :class:`mindspore.ops.IsFinite`.
  137. * - mindspore.ops.isinstance\_
  138. - Refer to :class:`mindspore.ops.IsInstance`.
  139. * - mindspore.ops.isnan
  140. - Refer to :class:`mindspore.ops.IsNan`.
  141. * - mindspore.ops.issubclass\_
  142. - Refer to :class:`mindspore.ops.IsSubClass`.
  143. * - mindspore.ops.log
  144. - Refer to :class:`mindspore.ops.Log`.
  145. * - mindspore.ops.logical_and
  146. - Refer to :class:`mindspore.ops.LogicalAnd`.
  147. * - mindspore.ops.le
  148. - Refer to :class:`mindspore.ops.LessEqual`.
  149. * - mindspore.ops.less
  150. - Refer to :class:`mindspore.ops.Less`.
  151. * - mindspore.ops.logical_and
  152. - Refer to :class:`mindspore.ops.LogicalAnd`.
  153. * - mindspore.ops.logical_not
  154. - Refer to :class:`mindspore.ops.LogicalNot`.
  155. * - mindspore.ops.logical_or
  156. - Refer to :class:`mindspore.ops.LogicalOr`.
  157. * - mindspore.ops.maximum
  158. - Refer to :class:`mindspore.ops.Maximum`.
  159. * - mindspore.ops.minimum
  160. - Refer to :class:`mindspore.ops.Minimum`.
  161. * - mindspore.ops.mul
  162. - Refer to :class:`mindspore.ops.Mul`.
  163. * - mindspore.ops.neg_tensor
  164. - Refer to :class:`mindspore.ops.Neg`.
  165. * - mindspore.ops.not_equal
  166. - Refer to :class:`mindspore.ops.NotEqual`.
  167. * - mindspore.ops.not_in_dict
  168. - Determine whether the object is not in the dict.
  169. * - mindspore.ops.ones_like
  170. - Refer to :class:`mindspore.ops.OnesLike`.
  171. * - mindspore.ops.partial
  172. - Refer to :class:`mindspore.ops.Partial`.
  173. * - mindspore.ops.pows
  174. - Refer to :class:`mindspore.ops.Pow`.
  175. * - mindspore.ops.print\_
  176. - Refer to :class:`mindspore.ops.Print`.
  177. * - mindspore.ops.rank
  178. - Refer to :class:`mindspore.ops.Rank`.
  179. * - mindspore.ops.reduce_max
  180. - Refer to :class:`mindspore.ops.ReduceMax`.
  181. * - mindspore.ops.reduce_mean
  182. - Refer to :class:`mindspore.ops.ReduceMean`.
  183. * - mindspore.ops.reduce_min
  184. - Refer to :class:`mindspore.ops.ReduceMin`.
  185. * - mindspore.ops.reduce_prod
  186. - Refer to :class:`mindspore.ops.ReduceProd`.
  187. * - mindspore.ops.reduce_sum
  188. - Refer to :class:`mindspore.ops.ReduceSum`.
  189. * - mindspore.ops.reshape
  190. - Refer to :class:`mindspore.ops.Reshape`.
  191. * - mindspore.ops.same_type_shape
  192. - Refer to :class:`mindspore.ops.SameTypeShape`.
  193. * - mindspore.ops.scalar_add
  194. - Get the sum of two numbers. (Usage is the same as "+" in Python)
  195. * - mindspore.ops.scalar_cast
  196. - Refer to :class:`mindspore.ops.ScalarCast`.
  197. * - mindspore.ops.scalar_div
  198. - Get the quotient of dividing the first input number by the second input number. (Usage is the same as "/" in Python)
  199. * - mindspore.ops.scalar_eq
  200. - Determine whether two numbers are equal. (Usage is the same as "==" in Python)
  201. * - mindspore.ops.scalar_floordiv
  202. - Divide the first input number by the second input number and round down to the closest integer. (Usage is the same as "//" in Python)
  203. * - mindspore.ops.scalar_ge
  204. - Determine whether the number is greater than or equal to another number. (Usage is the same as ">=" in Python)
  205. * - mindspore.ops.scalar_gt
  206. - Determine whether the number is greater than another number. (Usage is the same as ">" in Python)
  207. * - mindspore.ops.scalar_le
  208. - Determine whether the number is less than or equal to another number. (Usage is the same as "<=" in Python)
  209. * - mindspore.ops.scalar_log
  210. - Get the natural logarithm of the input number.
  211. * - mindspore.ops.scalar_lt
  212. - Determine whether the number is less than another number. (Usage is the same as "<" in Python)
  213. * - mindspore.ops.scalar_mod
  214. - Get the remainder of dividing the first input number by the second input number. (Usage is the same as "%" in Python)
  215. * - mindspore.ops.scalar_mul
  216. - Get the product of the input two numbers. (Usage is the same as "*" in Python)
  217. * - mindspore.ops.scalar_ne
  218. - Determine whether two numbers are not equal. (Usage is the same as "!=" in Python)
  219. * - mindspore.ops.scalar_pow
  220. - Compute a number to the power of the second input number.
  221. * - mindspore.ops.scalar_sub
  222. - Subtract the second input number from the first input number. (Usage is the same as "-" in Python)
  223. * - mindspore.ops.scalar_to_array
  224. - Refer to :class:`mindspore.ops.ScalarToArray`.
  225. * - mindspore.ops.scalar_to_tensor
  226. - Refer to :class:`mindspore.ops.ScalarToTensor`.
  227. * - mindspore.ops.scalar_uadd
  228. - Get the positive value of the input number.
  229. * - mindspore.ops.scalar_usub
  230. - Get the negative value of the input number.
  231. * - mindspore.ops.scatter_nd
  232. - Refer to :class:`mindspore.ops.ScatterNd`.
  233. * - mindspore.ops.scatter_nd_update
  234. - Refer to :class:`mindspore.ops.ScatterNdUpdate`.
  235. * - mindspore.ops.scatter_update
  236. - Refer to :class:`mindspore.ops.ScatterUpdate`.
  237. * - mindspore.ops.shape
  238. - Refer to :class:`mindspore.ops.Shape`.
  239. * - mindspore.ops.shape_mul
  240. - The input of shape_mul must be shape multiply elements in tuple(shape).
  241. * - mindspore.ops.sin
  242. - Refer to :class:`mindspore.ops.Sin`.
  243. * - mindspore.ops.sinh
  244. - Refer to :class:`mindspore.ops.Sinh`.
  245. * - mindspore.ops.size
  246. - Refer to :class:`mindspore.ops.Size`.
  247. * - mindspore.ops.sort
  248. - Refer to :class:`mindspore.ops.Sort`.
  249. * - mindspore.ops.sqrt
  250. - Refer to :class:`mindspore.ops.Sqrt`.
  251. * - mindspore.ops.square
  252. - Refer to :class:`mindspore.ops.Square`.
  253. * - mindspore.ops.squeeze
  254. - Refer to :class:`mindspore.ops.Squeeze`.
  255. * - mindspore.ops.stack
  256. - Refer to :class:`mindspore.ops.Stack`.
  257. * - mindspore.ops.stop_gradient
  258. - Disable update during back propagation. (`stop_gradient <https://www.mindspore.cn/tutorials/en/master/autograd.html#stop-gradient>`_)
  259. * - mindspore.ops.strided_slice
  260. - Refer to :class:`mindspore.ops.StridedSlice`.
  261. * - mindspore.ops.string_concat
  262. - Concatenate two strings.
  263. * - mindspore.ops.string_eq
  264. - Determine if two strings are equal.
  265. * - mindspore.ops.sub
  266. - Refer to :class:`mindspore.ops.Sub`.
  267. * - mindspore.ops.tan
  268. - Refer to :class:`mindspore.ops.Tan`.
  269. * - mindspore.ops.tanh
  270. - Refer to :class:`mindspore.ops.Tanh`.
  271. * - mindspore.ops.tensor_add
  272. - Refer to :class:`mindspore.ops.Add`.
  273. * - mindspore.ops.tensor_div
  274. - Refer to :class:`mindspore.ops.RealDiv`.
  275. * - mindspore.ops.tensor_exp
  276. - Refer to :class:`mindspore.ops.Exp`.
  277. * - mindspore.ops.tensor_expm1
  278. - Refer to :class:`mindspore.ops.Expm1`.
  279. * - mindspore.ops.tensor_floordiv
  280. - Refer to :class:`mindspore.ops.FloorDiv`.
  281. * - mindspore.ops.tensor_ge
  282. - Refer to :class:`mindspore.ops.GreaterEqual`.
  283. * - mindspore.ops.tensor_gt
  284. - Refer to :class:`mindspore.ops.Greater`.
  285. * - mindspore.ops.tensor_le
  286. - Refer to :class:`mindspore.ops.LessEqual`.
  287. * - mindspore.ops.tensor_lt
  288. - Refer to :class:`mindspore.ops.Less`.
  289. * - mindspore.ops.tensor_mod
  290. - Refer to :class:`mindspore.ops.FloorMod`.
  291. * - mindspore.ops.tensor_mul
  292. - Refer to :class:`mindspore.ops.Mul`.
  293. * - mindspore.ops.tensor_pow
  294. - Refer to :class:`mindspore.ops.Pow`.
  295. * - mindspore.ops.tensor_scatter_add
  296. - Refer to :class:`mindspore.ops.TensorScatterAdd`.
  297. * - mindspore.ops.tensor_scatter_update
  298. - Refer to :class:`mindspore.ops.TensorScatterUpdate`.
  299. * - mindspore.ops.tensor_slice
  300. - Refer to :class:`mindspore.ops.Slice`.
  301. * - mindspore.ops.tensor_sub
  302. - Refer to :class:`mindspore.ops.Sub`.
  303. * - mindspore.ops.tile
  304. - Refer to :class:`mindspore.ops.Tile`.
  305. * - mindspore.ops.transpose
  306. - Refer to :class:`mindspore.ops.Transpose`.
  307. * - mindspore.ops.tuple_to_array
  308. - Refer to :class:`mindspore.ops.TupleToArray`.
  309. * - mindspore.ops.typeof
  310. - Get type of object.
  311. * - mindspore.ops.zeros_like
  312. - Refer to :class:`mindspore.ops.ZerosLike`.
  313. .. mscnplatformautosummary::
  314. :toctree: ops
  315. :nosignatures:
  316. :template: classtemplate.rst
  317. mindspore.ops.batch_dot
  318. mindspore.ops.clip_by_global_norm
  319. mindspore.ops.clip_by_value
  320. mindspore.ops.core
  321. mindspore.ops.count_nonzero
  322. mindspore.ops.cummin
  323. mindspore.ops.dot
  324. mindspore.ops.gamma
  325. mindspore.ops.grad
  326. mindspore.ops.GradOperation
  327. mindspore.ops.HyperMap
  328. mindspore.ops.jvp
  329. mindspore.ops.laplace
  330. mindspore.ops.Map
  331. mindspore.ops.matmul_
  332. mindspore.ops.multinomial
  333. mindspore.ops.MultitypeFuncGraph
  334. mindspore.ops.narrow
  335. mindspore.ops.normal
  336. mindspore.ops.poisson
  337. mindspore.ops.repeat_elements
  338. mindspore.ops.select
  339. mindspore.ops.sequence_mask
  340. mindspore.ops.tensor_dot
  341. mindspore.ops.uniform
  342. mindspore.ops.vjp
  343. 原语
  344. -----
  345. .. mscnautosummary::
  346. :toctree: ops
  347. :nosignatures:
  348. :template: classtemplate.rst
  349. mindspore.ops.constexpr
  350. mindspore.ops.prim_attr_register
  351. mindspore.ops.Primitive
  352. mindspore.ops.PrimitiveWithCheck
  353. mindspore.ops.PrimitiveWithInfer
  354. 函数实现注册
  355. ----------------
  356. .. mscnautosummary::
  357. :toctree: ops
  358. :nosignatures:
  359. :template: classtemplate.rst
  360. mindspore.ops.get_vm_impl_fn
  361. 算子信息注册
  362. ----------------
  363. .. mscnautosummary::
  364. :toctree: ops
  365. :nosignatures:
  366. :template: classtemplate.rst
  367. mindspore.ops.AiCPURegOp
  368. mindspore.ops.custom_info_register
  369. mindspore.ops.CustomRegOp
  370. mindspore.ops.DataType
  371. mindspore.ops.op_info_register
  372. mindspore.ops.TBERegOp