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
4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402
  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.arange
  318. mindspore.ops.batch_dot
  319. mindspore.ops.clip_by_global_norm
  320. mindspore.ops.clip_by_value
  321. mindspore.ops.core
  322. mindspore.ops.count_nonzero
  323. mindspore.ops.cummin
  324. mindspore.ops.dot
  325. mindspore.ops.gamma
  326. mindspore.ops.grad
  327. mindspore.ops.GradOperation
  328. mindspore.ops.HyperMap
  329. mindspore.ops.jvp
  330. mindspore.ops.laplace
  331. mindspore.ops.Map
  332. mindspore.ops.matmul
  333. mindspore.ops.multinomial
  334. mindspore.ops.MultitypeFuncGraph
  335. mindspore.ops.narrow
  336. mindspore.ops.normal
  337. mindspore.ops.poisson
  338. mindspore.ops.repeat_elements
  339. mindspore.ops.select
  340. mindspore.ops.sequence_mask
  341. mindspore.ops.tensor_dot
  342. mindspore.ops.uniform
  343. mindspore.ops.vjp
  344. 原语
  345. -----
  346. .. mscnautosummary::
  347. :toctree: ops
  348. :nosignatures:
  349. :template: classtemplate.rst
  350. mindspore.ops.constexpr
  351. mindspore.ops.prim_attr_register
  352. mindspore.ops.Primitive
  353. mindspore.ops.PrimitiveWithCheck
  354. mindspore.ops.PrimitiveWithInfer
  355. 函数实现注册
  356. ----------------
  357. .. mscnautosummary::
  358. :toctree: ops
  359. :nosignatures:
  360. :template: classtemplate.rst
  361. mindspore.ops.get_vm_impl_fn
  362. 算子信息注册
  363. ----------------
  364. .. mscnautosummary::
  365. :toctree: ops
  366. :nosignatures:
  367. :template: classtemplate.rst
  368. mindspore.ops.AiCPURegOp
  369. mindspore.ops.custom_info_register
  370. mindspore.ops.CustomRegOp
  371. mindspore.ops.DataType
  372. mindspore.ops.op_info_register
  373. mindspore.ops.TBERegOp