mindspore 2.4.10__cp311-cp311-win_amd64.whl → 2.6.0__cp311-cp311-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of mindspore might be problematic. Click here for more details.

Files changed (602) hide show
  1. mindspore/.commit_id +1 -1
  2. mindspore/Microsoft.VisualStudio.Telemetry.dll +0 -0
  3. mindspore/Newtonsoft.Json.dll +0 -0
  4. mindspore/__init__.py +13 -6
  5. mindspore/_c_dataengine.cp311-win_amd64.pyd +0 -0
  6. mindspore/_c_expression.cp311-win_amd64.pyd +0 -0
  7. mindspore/_c_mindrecord.cp311-win_amd64.pyd +0 -0
  8. mindspore/_check_jit_forbidden_api.py +3 -0
  9. mindspore/_checkparam.py +3 -38
  10. mindspore/_deprecated/__init__.py +17 -0
  11. mindspore/_deprecated/jit.py +198 -0
  12. mindspore/_extends/builtin_operations.py +1 -1
  13. mindspore/_extends/parallel_compile/akg_compiler/gen_custom_op_files.py +1 -1
  14. mindspore/_extends/parse/__init__.py +6 -7
  15. mindspore/_extends/parse/compile_config.py +83 -0
  16. mindspore/_extends/parse/deprecated/__init__.py +0 -0
  17. mindspore/_extends/parse/deprecated/deprecated_tensor_method.py +394 -0
  18. mindspore/_extends/parse/jit_fallback_modules/__init__.py +0 -0
  19. mindspore/_extends/parse/jit_fallback_modules/check_utils.py +123 -0
  20. mindspore/_extends/parse/jit_fallback_modules/third_party_modules.py +50 -0
  21. mindspore/_extends/parse/parser.py +47 -198
  22. mindspore/_extends/parse/resources.py +1 -5
  23. mindspore/_extends/parse/standard_method.py +229 -99
  24. mindspore/_extends/pijit/__init__.py +2 -2
  25. mindspore/_extends/pijit/pijit_func_white_list.py +17 -12
  26. mindspore/_extends/pijit/tensor_func_list.py +27 -0
  27. mindspore/_extends/utils.py +1 -1
  28. mindspore/amp.py +11 -5
  29. mindspore/atlprov.dll +0 -0
  30. mindspore/avcodec-59.dll +0 -0
  31. mindspore/avdevice-59.dll +0 -0
  32. mindspore/avfilter-8.dll +0 -0
  33. mindspore/avformat-59.dll +0 -0
  34. mindspore/avutil-57.dll +0 -0
  35. mindspore/boost/__init__.py +2 -2
  36. mindspore/boost/base.py +3 -7
  37. mindspore/boost/boost_cell_wrapper.py +138 -43
  38. mindspore/c1.dll +0 -0
  39. mindspore/c1xx.dll +0 -0
  40. mindspore/c2.dll +0 -0
  41. mindspore/common/__init__.py +6 -3
  42. mindspore/common/_grad_function.py +56 -0
  43. mindspore/common/_pijit_context.py +14 -5
  44. mindspore/common/_register_for_tensor.py +1 -2
  45. mindspore/common/_stub_tensor.py +30 -14
  46. mindspore/common/_tensor_cpp_method.py +17 -0
  47. mindspore/common/_tensor_docs.py +4760 -0
  48. mindspore/common/api.py +480 -372
  49. mindspore/common/auto_dynamic_shape.py +41 -44
  50. mindspore/common/dtype.py +39 -36
  51. mindspore/common/dump.py +9 -6
  52. mindspore/common/file_system.py +9 -1
  53. mindspore/common/generator.py +5 -0
  54. mindspore/common/hook_handle.py +6 -2
  55. mindspore/common/initializer.py +13 -10
  56. mindspore/common/jit_begin_end.py +94 -0
  57. mindspore/common/jit_config.py +6 -1
  58. mindspore/common/jit_context.py +76 -0
  59. mindspore/common/jit_trace.py +378 -0
  60. mindspore/common/lazy_inline.py +9 -3
  61. mindspore/common/mindir_util.py +10 -2
  62. mindspore/common/mutable.py +5 -4
  63. mindspore/common/parameter.py +135 -52
  64. mindspore/common/seed.py +2 -2
  65. mindspore/common/sparse_tensor.py +23 -17
  66. mindspore/common/tensor.py +975 -1981
  67. mindspore/communication/__init__.py +7 -5
  68. mindspore/communication/_comm_helper.py +52 -2
  69. mindspore/communication/comm_func.py +240 -181
  70. mindspore/communication/management.py +95 -26
  71. mindspore/context.py +324 -573
  72. mindspore/dataset/__init__.py +65 -37
  73. mindspore/dataset/audio/__init__.py +2 -8
  74. mindspore/dataset/audio/transforms.py +3 -17
  75. mindspore/dataset/callback/ds_callback.py +2 -1
  76. mindspore/dataset/core/config.py +87 -6
  77. mindspore/dataset/engine/cache_admin.py +3 -3
  78. mindspore/dataset/engine/cache_client.py +6 -5
  79. mindspore/dataset/engine/datasets.py +292 -267
  80. mindspore/dataset/engine/datasets_audio.py +22 -8
  81. mindspore/dataset/engine/datasets_standard_format.py +46 -27
  82. mindspore/dataset/engine/datasets_text.py +78 -48
  83. mindspore/dataset/engine/datasets_user_defined.py +183 -117
  84. mindspore/dataset/engine/datasets_vision.py +120 -44
  85. mindspore/dataset/engine/iterators.py +283 -63
  86. mindspore/dataset/engine/obs/obs_mindrecord_dataset.py +1 -1
  87. mindspore/dataset/engine/obs/util.py +8 -0
  88. mindspore/dataset/engine/queue.py +40 -0
  89. mindspore/dataset/engine/samplers.py +289 -43
  90. mindspore/dataset/engine/serializer_deserializer.py +3 -2
  91. mindspore/dataset/engine/validators.py +53 -11
  92. mindspore/dataset/text/__init__.py +7 -6
  93. mindspore/dataset/text/transforms.py +6 -5
  94. mindspore/dataset/text/utils.py +3 -3
  95. mindspore/dataset/transforms/__init__.py +0 -9
  96. mindspore/dataset/transforms/py_transforms_util.py +17 -0
  97. mindspore/dataset/transforms/transforms.py +31 -14
  98. mindspore/dataset/utils/browse_dataset.py +1 -1
  99. mindspore/dataset/vision/__init__.py +2 -9
  100. mindspore/dataset/vision/transforms.py +202 -158
  101. mindspore/dataset/vision/utils.py +7 -5
  102. mindspore/dataset/vision/validators.py +1 -2
  103. mindspore/device_context/__init__.py +21 -0
  104. mindspore/device_context/ascend/__init__.py +25 -0
  105. mindspore/device_context/ascend/device.py +72 -0
  106. mindspore/device_context/ascend/op_debug.py +153 -0
  107. mindspore/device_context/ascend/op_precision.py +193 -0
  108. mindspore/device_context/ascend/op_tuning.py +123 -0
  109. mindspore/{ops_generate/gen_constants.py → device_context/cpu/__init__.py} +6 -17
  110. mindspore/device_context/cpu/device.py +62 -0
  111. mindspore/device_context/cpu/op_tuning.py +43 -0
  112. mindspore/device_context/gpu/__init__.py +21 -0
  113. mindspore/device_context/gpu/device.py +70 -0
  114. mindspore/device_context/gpu/op_precision.py +67 -0
  115. mindspore/device_context/gpu/op_tuning.py +175 -0
  116. mindspore/device_manager.py +170 -0
  117. mindspore/dnnl.dll +0 -0
  118. mindspore/dpcmi.dll +0 -0
  119. mindspore/experimental/es/embedding_service.py +35 -27
  120. mindspore/experimental/llm_boost/__init__.py +1 -0
  121. mindspore/experimental/llm_boost/ascend_native/__init__.py +22 -0
  122. mindspore/experimental/llm_boost/ascend_native/llama_boost_ascend_native.py +209 -0
  123. mindspore/experimental/llm_boost/ascend_native/llm_boost.py +52 -0
  124. mindspore/experimental/llm_boost/atb/boost_base.py +2 -3
  125. mindspore/experimental/llm_boost/atb/llama_boost.py +6 -1
  126. mindspore/experimental/llm_boost/register.py +1 -0
  127. mindspore/experimental/map_parameter.py +4 -4
  128. mindspore/experimental/optim/adadelta.py +6 -6
  129. mindspore/experimental/optim/adagrad.py +4 -4
  130. mindspore/experimental/optim/adam.py +7 -0
  131. mindspore/experimental/optim/adamax.py +4 -4
  132. mindspore/experimental/optim/adamw.py +4 -0
  133. mindspore/experimental/optim/asgd.py +1 -1
  134. mindspore/experimental/optim/lr_scheduler.py +73 -46
  135. mindspore/experimental/optim/radam.py +34 -31
  136. mindspore/experimental/optim/rprop.py +1 -1
  137. mindspore/experimental/optim/sgd.py +1 -1
  138. mindspore/hal/contiguous_tensors_handle.py +6 -10
  139. mindspore/hal/device.py +55 -53
  140. mindspore/hal/event.py +52 -52
  141. mindspore/hal/memory.py +179 -120
  142. mindspore/hal/stream.py +150 -109
  143. mindspore/include/api/context.h +0 -1
  144. mindspore/include/dataset/constants.h +7 -4
  145. mindspore/include/dataset/execute.h +2 -2
  146. mindspore/jpeg62.dll +0 -0
  147. mindspore/log.py +50 -0
  148. mindspore/mindrecord/__init__.py +21 -8
  149. mindspore/mindrecord/config.py +17 -316
  150. mindspore/mindrecord/filereader.py +1 -9
  151. mindspore/mindrecord/filewriter.py +5 -15
  152. mindspore/mindrecord/mindpage.py +1 -9
  153. mindspore/mindspore_backend_common.dll +0 -0
  154. mindspore/mindspore_backend_manager.dll +0 -0
  155. mindspore/mindspore_common.dll +0 -0
  156. mindspore/mindspore_core.dll +0 -0
  157. mindspore/mindspore_dump.dll +0 -0
  158. mindspore/mindspore_frontend.dll +0 -0
  159. mindspore/mindspore_glog.dll +0 -0
  160. mindspore/mindspore_memory_pool.dll +0 -0
  161. mindspore/mindspore_ms_backend.dll +0 -0
  162. mindspore/mindspore_ops.dll +0 -0
  163. mindspore/{mindspore_backend.dll → mindspore_ops_host.dll} +0 -0
  164. mindspore/mindspore_ops_kernel_common.dll +0 -0
  165. mindspore/mindspore_profiler.dll +0 -0
  166. mindspore/mindspore_pyboost.dll +0 -0
  167. mindspore/mindspore_pynative.dll +0 -0
  168. mindspore/mindspore_res_manager.dll +0 -0
  169. mindspore/mindspore_runtime_pipeline.dll +0 -0
  170. mindspore/mint/__init__.py +798 -761
  171. mindspore/mint/distributed/__init__.py +70 -4
  172. mindspore/mint/distributed/distributed.py +2679 -44
  173. mindspore/mint/linalg/__init__.py +8 -0
  174. mindspore/mint/nn/__init__.py +743 -22
  175. mindspore/mint/nn/functional.py +716 -23
  176. mindspore/mint/nn/layer/__init__.py +21 -4
  177. mindspore/mint/nn/layer/_functions.py +334 -0
  178. mindspore/mint/nn/layer/activation.py +276 -1
  179. mindspore/mint/nn/layer/basic.py +123 -0
  180. mindspore/mint/nn/layer/conv.py +933 -0
  181. mindspore/mint/nn/layer/normalization.py +223 -28
  182. mindspore/mint/nn/layer/padding.py +797 -0
  183. mindspore/mint/nn/layer/pooling.py +235 -0
  184. mindspore/mint/optim/__init__.py +3 -1
  185. mindspore/mint/optim/adam.py +223 -0
  186. mindspore/mint/optim/adamw.py +26 -19
  187. mindspore/mint/optim/sgd.py +171 -0
  188. mindspore/mint/special/__init__.py +2 -1
  189. mindspore/msobj140.dll +0 -0
  190. mindspore/mspdb140.dll +0 -0
  191. mindspore/mspdbcore.dll +0 -0
  192. mindspore/mspdbst.dll +0 -0
  193. mindspore/mspft140.dll +0 -0
  194. mindspore/msvcdis140.dll +0 -0
  195. mindspore/msvcp140_1.dll +0 -0
  196. mindspore/msvcp140_2.dll +0 -0
  197. mindspore/msvcp140_atomic_wait.dll +0 -0
  198. mindspore/msvcp140_codecvt_ids.dll +0 -0
  199. mindspore/multiprocessing/__init__.py +5 -0
  200. mindspore/nn/__init__.py +4 -1
  201. mindspore/nn/cell.py +1373 -192
  202. mindspore/nn/dynamic_lr.py +2 -1
  203. mindspore/nn/layer/activation.py +29 -27
  204. mindspore/nn/layer/basic.py +51 -35
  205. mindspore/nn/layer/channel_shuffle.py +3 -3
  206. mindspore/nn/layer/container.py +1 -1
  207. mindspore/nn/layer/conv.py +53 -42
  208. mindspore/nn/layer/embedding.py +12 -11
  209. mindspore/nn/layer/normalization.py +56 -49
  210. mindspore/nn/layer/padding.py +4 -3
  211. mindspore/nn/layer/pooling.py +120 -42
  212. mindspore/nn/layer/rnn_cells.py +1 -1
  213. mindspore/nn/layer/rnns.py +2 -1
  214. mindspore/nn/layer/timedistributed.py +5 -5
  215. mindspore/nn/layer/transformer.py +59 -36
  216. mindspore/nn/learning_rate_schedule.py +8 -4
  217. mindspore/nn/loss/loss.py +58 -55
  218. mindspore/nn/optim/ada_grad.py +7 -5
  219. mindspore/nn/optim/adadelta.py +11 -9
  220. mindspore/nn/optim/adafactor.py +1 -1
  221. mindspore/nn/optim/adam.py +19 -15
  222. mindspore/nn/optim/adamax.py +8 -7
  223. mindspore/nn/optim/adasum.py +5 -5
  224. mindspore/nn/optim/asgd.py +3 -1
  225. mindspore/nn/optim/ftrl.py +11 -9
  226. mindspore/nn/optim/lamb.py +1 -1
  227. mindspore/nn/optim/lars.py +1 -4
  228. mindspore/nn/optim/lazyadam.py +12 -10
  229. mindspore/nn/optim/momentum.py +7 -6
  230. mindspore/nn/optim/optimizer.py +3 -3
  231. mindspore/nn/optim/proximal_ada_grad.py +12 -10
  232. mindspore/nn/optim/rmsprop.py +13 -12
  233. mindspore/nn/optim/rprop.py +11 -9
  234. mindspore/nn/optim/sgd.py +9 -6
  235. mindspore/nn/optim/tft_wrapper.py +5 -2
  236. mindspore/nn/optim/thor.py +2 -1
  237. mindspore/nn/probability/bijector/bijector.py +17 -11
  238. mindspore/nn/probability/bijector/gumbel_cdf.py +5 -5
  239. mindspore/nn/probability/bijector/invert.py +2 -2
  240. mindspore/nn/probability/bijector/scalar_affine.py +3 -3
  241. mindspore/nn/probability/bijector/softplus.py +3 -2
  242. mindspore/nn/probability/distribution/beta.py +3 -3
  243. mindspore/nn/probability/distribution/categorical.py +1 -1
  244. mindspore/nn/probability/distribution/cauchy.py +4 -2
  245. mindspore/nn/probability/distribution/exponential.py +6 -7
  246. mindspore/nn/probability/distribution/gamma.py +2 -2
  247. mindspore/nn/probability/distribution/gumbel.py +2 -2
  248. mindspore/nn/probability/distribution/half_normal.py +5 -3
  249. mindspore/nn/probability/distribution/logistic.py +5 -3
  250. mindspore/nn/probability/distribution/poisson.py +1 -1
  251. mindspore/nn/probability/distribution/uniform.py +5 -3
  252. mindspore/nn/reinforcement/_tensors_queue.py +1 -1
  253. mindspore/nn/reinforcement/tensor_array.py +1 -1
  254. mindspore/nn/utils/init.py +13 -11
  255. mindspore/nn/wrap/__init__.py +6 -6
  256. mindspore/nn/wrap/cell_wrapper.py +181 -122
  257. mindspore/nn/wrap/grad_reducer.py +45 -36
  258. mindspore/nn/wrap/loss_scale.py +6 -7
  259. mindspore/numpy/array_creations.py +63 -65
  260. mindspore/numpy/array_ops.py +149 -144
  261. mindspore/numpy/logic_ops.py +41 -42
  262. mindspore/numpy/math_ops.py +361 -359
  263. mindspore/numpy/utils.py +17 -18
  264. mindspore/numpy/utils_const.py +5 -6
  265. mindspore/opencv_core452.dll +0 -0
  266. mindspore/opencv_imgcodecs452.dll +0 -0
  267. mindspore/opencv_imgproc452.dll +0 -0
  268. mindspore/ops/__init__.py +5 -3
  269. mindspore/ops/_grad_experimental/grad_comm_ops.py +112 -16
  270. mindspore/ops/_grad_experimental/grad_debug_ops.py +14 -2
  271. mindspore/ops/_grad_experimental/grad_inner_ops.py +9 -0
  272. mindspore/ops/_grad_experimental/grad_math_ops.py +2 -1
  273. mindspore/ops/_grad_experimental/taylor_rule.py +29 -0
  274. mindspore/ops/_op_impl/cpu/__init__.py +1 -0
  275. mindspore/ops/_op_impl/cpu/raise_op.py +28 -0
  276. mindspore/ops/_register_for_op.py +0 -11
  277. mindspore/{ops_generate → ops/_utils}/arg_dtype_cast.py +123 -4
  278. mindspore/{ops_generate → ops/_utils}/arg_handler.py +3 -65
  279. mindspore/ops/_vmap/vmap_array_ops.py +52 -25
  280. mindspore/ops/_vmap/vmap_base.py +0 -2
  281. mindspore/ops/_vmap/vmap_grad_nn_ops.py +21 -14
  282. mindspore/ops/_vmap/vmap_math_ops.py +15 -16
  283. mindspore/ops/_vmap/vmap_nn_ops.py +29 -42
  284. mindspore/ops/auto_generate/__init__.py +4 -3
  285. mindspore/ops/auto_generate/cpp_create_prim_instance_helper.py +258 -46
  286. mindspore/ops/auto_generate/gen_extend_func.py +757 -185
  287. mindspore/ops/auto_generate/gen_ops_def.py +4197 -2243
  288. mindspore/ops/auto_generate/gen_ops_prim.py +16976 -6055
  289. mindspore/ops/auto_generate/pyboost_inner_prim.py +221 -87
  290. mindspore/ops/composite/__init__.py +2 -1
  291. mindspore/ops/composite/base.py +20 -25
  292. mindspore/ops/composite/math_ops.py +6 -16
  293. mindspore/ops/composite/multitype_ops/__init__.py +5 -2
  294. mindspore/ops/composite/multitype_ops/_compile_utils.py +228 -30
  295. mindspore/ops/composite/multitype_ops/_constexpr_utils.py +1 -2
  296. mindspore/ops/composite/multitype_ops/add_impl.py +2 -1
  297. mindspore/ops/composite/multitype_ops/bitwise_and_impl.py +2 -1
  298. mindspore/ops/composite/multitype_ops/bitwise_or_impl.py +2 -1
  299. mindspore/ops/composite/multitype_ops/bitwise_xor_impl.py +2 -1
  300. mindspore/ops/composite/multitype_ops/div_impl.py +6 -4
  301. mindspore/ops/composite/multitype_ops/equal_impl.py +4 -3
  302. mindspore/ops/composite/multitype_ops/floordiv_impl.py +2 -1
  303. mindspore/ops/composite/multitype_ops/getitem_impl.py +3 -2
  304. mindspore/ops/composite/multitype_ops/greater_equal_impl.py +4 -3
  305. mindspore/ops/composite/multitype_ops/greater_impl.py +4 -3
  306. mindspore/ops/composite/multitype_ops/in_impl.py +2 -1
  307. mindspore/ops/composite/multitype_ops/invert_impl.py +50 -0
  308. mindspore/ops/composite/multitype_ops/left_shift_impl.py +2 -1
  309. mindspore/ops/composite/multitype_ops/less_equal_impl.py +4 -3
  310. mindspore/ops/composite/multitype_ops/less_impl.py +4 -3
  311. mindspore/ops/composite/multitype_ops/logic_not_impl.py +3 -2
  312. mindspore/ops/composite/multitype_ops/logical_and_impl.py +2 -1
  313. mindspore/ops/composite/multitype_ops/logical_or_impl.py +2 -1
  314. mindspore/ops/composite/multitype_ops/mod_impl.py +2 -1
  315. mindspore/ops/composite/multitype_ops/mul_impl.py +3 -2
  316. mindspore/ops/composite/multitype_ops/negative_impl.py +2 -1
  317. mindspore/ops/composite/multitype_ops/not_equal_impl.py +2 -1
  318. mindspore/ops/composite/multitype_ops/not_in_impl.py +2 -1
  319. mindspore/ops/composite/multitype_ops/ones_like_impl.py +18 -0
  320. mindspore/ops/composite/multitype_ops/pow_impl.py +2 -30
  321. mindspore/ops/composite/multitype_ops/right_shift_impl.py +2 -1
  322. mindspore/ops/composite/multitype_ops/setitem_impl.py +2 -1
  323. mindspore/ops/composite/multitype_ops/sub_impl.py +2 -1
  324. mindspore/ops/function/__init__.py +40 -2
  325. mindspore/ops/function/_add_attr_func.py +58 -0
  326. mindspore/ops/function/array_func.py +2089 -2403
  327. mindspore/ops/function/clip_func.py +80 -23
  328. mindspore/ops/function/debug_func.py +57 -57
  329. mindspore/ops/function/grad/__init__.py +1 -0
  330. mindspore/ops/function/grad/grad_func.py +104 -71
  331. mindspore/ops/function/image_func.py +2 -2
  332. mindspore/ops/function/linalg_func.py +47 -78
  333. mindspore/ops/function/math_func.py +4351 -3813
  334. mindspore/ops/function/nn_func.py +1712 -637
  335. mindspore/ops/function/other_func.py +159 -1
  336. mindspore/ops/function/parameter_func.py +18 -84
  337. mindspore/ops/function/random_func.py +452 -387
  338. mindspore/ops/function/reshard_func.py +4 -70
  339. mindspore/ops/function/sparse_func.py +3 -3
  340. mindspore/ops/function/sparse_unary_func.py +6 -6
  341. mindspore/ops/function/spectral_func.py +25 -58
  342. mindspore/ops/function/vmap_func.py +26 -18
  343. mindspore/ops/functional.py +23 -7
  344. mindspore/ops/functional_overload.py +1548 -0
  345. mindspore/ops/op_info_register.py +32 -244
  346. mindspore/ops/operations/__init__.py +23 -15
  347. mindspore/ops/operations/_custom_ops_utils.py +235 -0
  348. mindspore/ops/operations/_embedding_cache_ops.py +4 -4
  349. mindspore/ops/operations/_grad_ops.py +2 -43
  350. mindspore/ops/operations/_infer_ops.py +2 -1
  351. mindspore/ops/operations/_inner_ops.py +43 -84
  352. mindspore/ops/operations/_ms_kernel.py +4 -10
  353. mindspore/ops/operations/_rl_inner_ops.py +1 -1
  354. mindspore/ops/operations/_scalar_ops.py +3 -2
  355. mindspore/ops/operations/_sequence_ops.py +1 -1
  356. mindspore/ops/operations/_tensor_array.py +1 -1
  357. mindspore/ops/operations/array_ops.py +81 -324
  358. mindspore/ops/operations/comm_ops.py +154 -108
  359. mindspore/ops/operations/custom_ops.py +298 -87
  360. mindspore/ops/operations/debug_ops.py +157 -59
  361. mindspore/ops/operations/inner_ops.py +7 -5
  362. mindspore/ops/operations/linalg_ops.py +1 -57
  363. mindspore/ops/operations/manually_defined/_inner.py +1 -1
  364. mindspore/ops/operations/manually_defined/ops_def.py +928 -180
  365. mindspore/ops/operations/math_ops.py +32 -234
  366. mindspore/ops/operations/nn_ops.py +212 -531
  367. mindspore/ops/operations/other_ops.py +62 -9
  368. mindspore/ops/operations/random_ops.py +13 -7
  369. mindspore/ops/operations/reshard_ops.py +1 -1
  370. mindspore/ops/operations/sparse_ops.py +2 -2
  371. mindspore/ops/primitive.py +66 -53
  372. mindspore/ops/tensor_method.py +1895 -0
  373. mindspore/ops_generate/__init__.py +0 -5
  374. mindspore/ops_generate/aclnn/__init__.py +0 -0
  375. mindspore/ops_generate/aclnn/aclnn_kernel_register_auto_cc_generator.py +135 -0
  376. mindspore/ops_generate/aclnn/gen_aclnn_implement.py +257 -0
  377. mindspore/ops_generate/api/__init__.py +0 -0
  378. mindspore/ops_generate/api/add_tensor_docs_generator.py +56 -0
  379. mindspore/ops_generate/api/cpp_create_prim_instance_helper_generator.py +105 -0
  380. mindspore/ops_generate/api/functional_map_cpp_generator.py +504 -0
  381. mindspore/ops_generate/api/functional_overload_py_generator.py +112 -0
  382. mindspore/ops_generate/api/functions_cc_generator.py +237 -0
  383. mindspore/ops_generate/api/gen_api.py +103 -0
  384. mindspore/ops_generate/api/op_api_proto.py +235 -0
  385. mindspore/ops_generate/api/tensor_func_reg_cpp_generator.py +461 -0
  386. mindspore/ops_generate/common/__init__.py +0 -0
  387. mindspore/ops_generate/common/base_generator.py +11 -0
  388. mindspore/ops_generate/common/gen_constants.py +91 -0
  389. mindspore/ops_generate/common/gen_utils.py +348 -0
  390. mindspore/ops_generate/common/op_proto.py +473 -0
  391. mindspore/ops_generate/common/template.py +523 -0
  392. mindspore/ops_generate/gen_ops.py +22 -1069
  393. mindspore/ops_generate/op_def/__init__.py +0 -0
  394. mindspore/ops_generate/op_def/gen_op_def.py +90 -0
  395. mindspore/ops_generate/op_def/lite_ops_cpp_generator.py +191 -0
  396. mindspore/ops_generate/op_def/ops_def_cc_generator.py +296 -0
  397. mindspore/ops_generate/op_def/ops_def_h_generator.py +74 -0
  398. mindspore/ops_generate/op_def/ops_name_h_generator.py +83 -0
  399. mindspore/ops_generate/op_def/ops_primitive_h_generator.py +125 -0
  400. mindspore/ops_generate/op_def_py/__init__.py +0 -0
  401. mindspore/ops_generate/op_def_py/gen_op_def_py.py +47 -0
  402. mindspore/ops_generate/op_def_py/op_def_py_generator.py +132 -0
  403. mindspore/ops_generate/op_def_py/op_prim_py_generator.py +489 -0
  404. mindspore/ops_generate/pyboost/__init__.py +0 -0
  405. mindspore/ops_generate/pyboost/auto_grad_impl_cc_generator.py +139 -0
  406. mindspore/ops_generate/pyboost/auto_grad_reg_cc_generator.py +93 -0
  407. mindspore/ops_generate/pyboost/gen_pyboost_func.py +175 -0
  408. mindspore/ops_generate/pyboost/op_template_parser.py +517 -0
  409. mindspore/ops_generate/pyboost/pyboost_functions_cpp_generator.py +407 -0
  410. mindspore/ops_generate/pyboost/pyboost_functions_h_generator.py +100 -0
  411. mindspore/ops_generate/pyboost/pyboost_functions_py_generator.py +148 -0
  412. mindspore/ops_generate/pyboost/pyboost_grad_function_cpp_generator.py +155 -0
  413. mindspore/ops_generate/pyboost/pyboost_inner_prim_generator.py +132 -0
  414. mindspore/ops_generate/pyboost/pyboost_native_grad_functions_generator.py +272 -0
  415. mindspore/ops_generate/pyboost/pyboost_op_cpp_code_generator.py +938 -0
  416. mindspore/ops_generate/pyboost/pyboost_overload_functions_cpp_generator.py +357 -0
  417. mindspore/ops_generate/{pyboost_utils.py → pyboost/pyboost_utils.py} +179 -36
  418. mindspore/ops_generate/resources/__init__.py +0 -0
  419. mindspore/ops_generate/resources/resource_list.py +30 -0
  420. mindspore/ops_generate/resources/resource_loader.py +36 -0
  421. mindspore/ops_generate/resources/resource_manager.py +64 -0
  422. mindspore/ops_generate/resources/yaml_loader.py +88 -0
  423. mindspore/ops_generate/tensor_py_cc_generator.py +122 -0
  424. mindspore/parallel/__init__.py +7 -3
  425. mindspore/parallel/_auto_parallel_context.py +159 -40
  426. mindspore/parallel/_cell_wrapper.py +132 -15
  427. mindspore/parallel/_parallel_serialization.py +107 -5
  428. mindspore/parallel/_ps_context.py +1 -1
  429. mindspore/parallel/_recovery_context.py +7 -2
  430. mindspore/parallel/_tensor.py +142 -18
  431. mindspore/parallel/_utils.py +199 -23
  432. mindspore/parallel/algo_parameter_config.py +4 -4
  433. mindspore/parallel/auto_parallel.py +732 -0
  434. mindspore/parallel/checkpoint_convert.py +159 -0
  435. mindspore/parallel/checkpoint_transform.py +700 -35
  436. mindspore/parallel/cluster/process_entity/_api.py +276 -50
  437. mindspore/parallel/cluster/process_entity/_utils.py +41 -6
  438. mindspore/parallel/cluster/run.py +21 -4
  439. mindspore/parallel/function/__init__.py +24 -0
  440. mindspore/parallel/function/reshard_func.py +258 -0
  441. mindspore/parallel/nn/__init__.py +25 -0
  442. mindspore/parallel/nn/parallel_cell_wrapper.py +263 -0
  443. mindspore/parallel/nn/parallel_grad_reducer.py +169 -0
  444. mindspore/parallel/parameter_broadcast.py +25 -14
  445. mindspore/parallel/shard.py +137 -59
  446. mindspore/parallel/transform_safetensors.py +364 -305
  447. mindspore/pgodb140.dll +0 -0
  448. mindspore/pgort140.dll +0 -0
  449. mindspore/profiler/__init__.py +22 -5
  450. mindspore/profiler/analysis/__init__.py +0 -0
  451. mindspore/profiler/analysis/parser/__init__.py +0 -0
  452. mindspore/profiler/analysis/parser/ascend_cann_parser.py +170 -0
  453. mindspore/profiler/analysis/parser/base_parser.py +158 -0
  454. mindspore/profiler/analysis/parser/framework_cann_relation_parser.py +45 -0
  455. mindspore/profiler/analysis/parser/ms_framework_parser.py +142 -0
  456. mindspore/profiler/analysis/parser/ms_minddata_parser.py +145 -0
  457. mindspore/profiler/analysis/parser/timeline_assembly_factory/__init__.py +0 -0
  458. mindspore/profiler/analysis/parser/timeline_assembly_factory/ascend_timeline_assembler.py +264 -0
  459. mindspore/profiler/analysis/parser/timeline_assembly_factory/base_timeline_assembler.py +40 -0
  460. mindspore/profiler/analysis/parser/timeline_assembly_factory/trace_view_container.py +109 -0
  461. mindspore/profiler/analysis/parser/timeline_creator/__init__.py +0 -0
  462. mindspore/profiler/analysis/parser/timeline_creator/base_timeline_creator.py +44 -0
  463. mindspore/profiler/analysis/parser/timeline_creator/cpu_op_timeline_creator.py +90 -0
  464. mindspore/profiler/analysis/parser/timeline_creator/fwk_timeline_creator.py +76 -0
  465. mindspore/profiler/analysis/parser/timeline_creator/msprof_timeline_creator.py +103 -0
  466. mindspore/profiler/analysis/parser/timeline_creator/scope_layer_timeline_creator.py +134 -0
  467. mindspore/profiler/analysis/parser/timeline_event/__init__.py +0 -0
  468. mindspore/profiler/analysis/parser/timeline_event/base_event.py +233 -0
  469. mindspore/profiler/analysis/parser/timeline_event/cpu_op_event.py +47 -0
  470. mindspore/profiler/analysis/parser/timeline_event/flow_event.py +36 -0
  471. mindspore/profiler/analysis/parser/timeline_event/fwk_event.py +415 -0
  472. mindspore/profiler/analysis/parser/timeline_event/msprof_event.py +73 -0
  473. mindspore/profiler/analysis/parser/timeline_event/scope_layer_event.py +53 -0
  474. mindspore/profiler/analysis/parser/timeline_event/timeline_event_pool.py +146 -0
  475. mindspore/profiler/analysis/task_manager.py +131 -0
  476. mindspore/profiler/analysis/time_converter.py +84 -0
  477. mindspore/profiler/analysis/viewer/__init__.py +0 -0
  478. mindspore/profiler/analysis/viewer/ascend_communication_viewer.py +372 -0
  479. mindspore/profiler/analysis/viewer/ascend_integrate_viewer.py +87 -0
  480. mindspore/profiler/analysis/viewer/ascend_kernel_details_viewer.py +250 -0
  481. mindspore/profiler/analysis/viewer/ascend_memory_viewer.py +320 -0
  482. mindspore/profiler/analysis/viewer/ascend_op_memory_viewer.py +327 -0
  483. mindspore/profiler/analysis/viewer/ascend_step_trace_time_viewer.py +376 -0
  484. mindspore/profiler/analysis/viewer/ascend_timeline_viewer.py +58 -0
  485. mindspore/profiler/analysis/viewer/base_viewer.py +26 -0
  486. mindspore/profiler/analysis/viewer/ms_dataset_viewer.py +96 -0
  487. mindspore/profiler/analysis/viewer/ms_minddata_viewer.py +581 -0
  488. mindspore/profiler/analysis/work_flow.py +73 -0
  489. mindspore/profiler/common/ascend_msprof_exporter.py +139 -0
  490. mindspore/profiler/common/command_executor.py +90 -0
  491. mindspore/profiler/common/constant.py +186 -3
  492. mindspore/profiler/common/file_manager.py +208 -0
  493. mindspore/profiler/common/log.py +130 -0
  494. mindspore/profiler/common/msprof_cmd_tool.py +221 -0
  495. mindspore/profiler/common/path_manager.py +395 -0
  496. mindspore/profiler/common/process_bar.py +168 -0
  497. mindspore/profiler/common/process_pool.py +9 -3
  498. mindspore/profiler/common/profiler_context.py +500 -0
  499. mindspore/profiler/common/profiler_info.py +304 -0
  500. mindspore/profiler/common/profiler_meta_data.py +74 -0
  501. mindspore/profiler/common/profiler_output_path.py +284 -0
  502. mindspore/profiler/common/profiler_parameters.py +251 -0
  503. mindspore/profiler/common/profiler_path_manager.py +179 -0
  504. mindspore/profiler/common/record_function.py +76 -0
  505. mindspore/profiler/common/tlv_decoder.py +76 -0
  506. mindspore/profiler/common/util.py +75 -2
  507. mindspore/profiler/dynamic_profiler.py +341 -75
  508. mindspore/profiler/envprofiler.py +163 -0
  509. mindspore/profiler/experimental_config.py +197 -0
  510. mindspore/profiler/mstx.py +242 -0
  511. mindspore/profiler/platform/__init__.py +21 -0
  512. mindspore/profiler/platform/base_profiler.py +40 -0
  513. mindspore/profiler/platform/cpu_profiler.py +124 -0
  514. mindspore/profiler/platform/gpu_profiler.py +74 -0
  515. mindspore/profiler/platform/npu_profiler.py +335 -0
  516. mindspore/profiler/profiler.py +1073 -90
  517. mindspore/profiler/profiler_action_controller.py +187 -0
  518. mindspore/profiler/profiler_interface.py +118 -0
  519. mindspore/profiler/schedule.py +243 -0
  520. mindspore/rewrite/api/node.py +15 -13
  521. mindspore/rewrite/api/symbol_tree.py +2 -3
  522. mindspore/run_check/_check_version.py +27 -20
  523. mindspore/run_check/run_check.py +1 -1
  524. mindspore/runtime/__init__.py +37 -0
  525. mindspore/runtime/device.py +27 -0
  526. mindspore/runtime/event.py +209 -0
  527. mindspore/runtime/executor.py +177 -0
  528. mindspore/runtime/memory.py +416 -0
  529. mindspore/runtime/stream.py +460 -0
  530. mindspore/runtime/thread_bind_core.py +401 -0
  531. mindspore/safeguard/rewrite_obfuscation.py +12 -9
  532. mindspore/swresample-4.dll +0 -0
  533. mindspore/swscale-6.dll +0 -0
  534. mindspore/tbbmalloc.dll +0 -0
  535. mindspore/tinyxml2.dll +0 -0
  536. mindspore/train/__init__.py +8 -8
  537. mindspore/train/_utils.py +96 -27
  538. mindspore/train/amp.py +9 -5
  539. mindspore/train/callback/__init__.py +2 -2
  540. mindspore/train/callback/_callback.py +2 -16
  541. mindspore/train/callback/_checkpoint.py +53 -55
  542. mindspore/train/callback/_cluster_monitor.py +14 -18
  543. mindspore/train/callback/_early_stop.py +1 -1
  544. mindspore/train/callback/_flops_collector.py +103 -68
  545. mindspore/train/callback/_history.py +8 -5
  546. mindspore/train/callback/_lambda_callback.py +2 -2
  547. mindspore/train/callback/_landscape.py +0 -3
  548. mindspore/train/callback/_loss_monitor.py +2 -1
  549. mindspore/train/callback/_on_request_exit.py +6 -5
  550. mindspore/train/callback/_reduce_lr_on_plateau.py +11 -6
  551. mindspore/train/callback/_summary_collector.py +52 -19
  552. mindspore/train/callback/_time_monitor.py +2 -1
  553. mindspore/train/callback/{_tft_register.py → _train_fault_tolerance.py} +228 -108
  554. mindspore/train/data_sink.py +25 -2
  555. mindspore/train/dataset_helper.py +15 -16
  556. mindspore/train/loss_scale_manager.py +8 -7
  557. mindspore/train/metrics/accuracy.py +3 -3
  558. mindspore/train/metrics/confusion_matrix.py +9 -9
  559. mindspore/train/metrics/error.py +3 -3
  560. mindspore/train/metrics/hausdorff_distance.py +4 -4
  561. mindspore/train/metrics/mean_surface_distance.py +3 -3
  562. mindspore/train/metrics/metric.py +0 -12
  563. mindspore/train/metrics/occlusion_sensitivity.py +4 -2
  564. mindspore/train/metrics/precision.py +11 -10
  565. mindspore/train/metrics/recall.py +9 -9
  566. mindspore/train/metrics/root_mean_square_surface_distance.py +2 -2
  567. mindspore/train/mind_ir_pb2.py +174 -46
  568. mindspore/train/model.py +269 -136
  569. mindspore/train/serialization.py +622 -978
  570. mindspore/train/summary/_summary_adapter.py +2 -2
  571. mindspore/train/summary/summary_record.py +2 -3
  572. mindspore/train/train_thor/model_thor.py +1 -1
  573. mindspore/turbojpeg.dll +0 -0
  574. mindspore/utils/__init__.py +6 -3
  575. mindspore/utils/dryrun.py +140 -0
  576. mindspore/utils/hooks.py +81 -0
  577. mindspore/utils/runtime_execution_order_check.py +552 -0
  578. mindspore/utils/utils.py +138 -4
  579. mindspore/vcmeta.dll +0 -0
  580. mindspore/vcruntime140.dll +0 -0
  581. mindspore/vcruntime140_1.dll +0 -0
  582. mindspore/version.py +1 -1
  583. {mindspore-2.4.10.dist-info → mindspore-2.6.0.dist-info}/METADATA +3 -3
  584. {mindspore-2.4.10.dist-info → mindspore-2.6.0.dist-info}/RECORD +587 -418
  585. {mindspore-2.4.10.dist-info → mindspore-2.6.0.dist-info}/entry_points.txt +1 -1
  586. mindspore/_install_custom.py +0 -43
  587. mindspore/common/_register_for_adapter.py +0 -74
  588. mindspore/common/_tensor_overload.py +0 -139
  589. mindspore/mindspore_np_dtype.dll +0 -0
  590. mindspore/ops/auto_generate/gen_arg_dtype_cast.py +0 -252
  591. mindspore/ops/auto_generate/gen_arg_handler.py +0 -197
  592. mindspore/ops/operations/_opaque_predicate_registry.py +0 -41
  593. mindspore/ops_generate/gen_aclnn_implement.py +0 -263
  594. mindspore/ops_generate/gen_ops_inner_prim.py +0 -131
  595. mindspore/ops_generate/gen_pyboost_func.py +0 -1052
  596. mindspore/ops_generate/gen_utils.py +0 -209
  597. mindspore/ops_generate/op_proto.py +0 -145
  598. mindspore/ops_generate/template.py +0 -261
  599. mindspore/profiler/envprofiling.py +0 -254
  600. mindspore/profiler/profiling.py +0 -1926
  601. {mindspore-2.4.10.dist-info → mindspore-2.6.0.dist-info}/WHEEL +0 -0
  602. {mindspore-2.4.10.dist-info → mindspore-2.6.0.dist-info}/top_level.txt +0 -0
@@ -125,12 +125,12 @@ class AdjustBrightness(ImageTensorOperation, PyTensorOperation):
125
125
 
126
126
  Args:
127
127
  brightness_factor (float): How much to adjust the brightness, must be non negative.
128
- ``0`` gives a black image, ``1`` gives the original image,
129
- while ``2`` increases the brightness by a factor of 2.
128
+ ``0.0`` gives a black image, ``1.0`` gives the original image,
129
+ while ``2.0`` increases the brightness by a factor of 2.
130
130
 
131
131
  Raises:
132
132
  TypeError: If `brightness_factor` is not of type float.
133
- ValueError: If `brightness_factor` is less than 0.
133
+ ValueError: If `brightness_factor` is less than ``0.0``.
134
134
  RuntimeError: If shape of the input image is not <H, W, C>.
135
135
 
136
136
  Supported Platforms:
@@ -173,15 +173,15 @@ class AdjustBrightness(ImageTensorOperation, PyTensorOperation):
173
173
  """
174
174
  Set the device for the current operator execution.
175
175
 
176
- - When the device is Ascend, input shape should be limited from [4, 6] to [8192, 4096].
176
+ When the device is Ascend, input shape should be limited from [4, 6] to [8192, 4096].
177
177
 
178
178
  Args:
179
179
  device_target (str, optional): The operator will be executed on this device. Currently supports
180
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
180
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
181
181
 
182
182
  Raises:
183
183
  TypeError: If `device_target` is not of type str.
184
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
184
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
185
185
 
186
186
  Supported Platforms:
187
187
  ``CPU`` ``Ascend``
@@ -239,12 +239,12 @@ class AdjustContrast(ImageTensorOperation, PyTensorOperation):
239
239
 
240
240
  Args:
241
241
  contrast_factor (float): How much to adjust the contrast, must be non negative.
242
- ``0`` gives a solid gray image, ``1`` gives the original image,
243
- while ``2`` increases the contrast by a factor of 2.
242
+ ``0.0`` gives a solid gray image, ``1.0`` gives the original image,
243
+ while ``2.0`` increases the contrast by a factor of 2.
244
244
 
245
245
  Raises:
246
246
  TypeError: If `contrast_factor` is not of type float.
247
- ValueError: If `contrast_factor` is less than 0.
247
+ ValueError: If `contrast_factor` is less than ``0.0``.
248
248
  RuntimeError: If shape of the input image is not <H, W, C>.
249
249
 
250
250
  Supported Platforms:
@@ -286,15 +286,15 @@ class AdjustContrast(ImageTensorOperation, PyTensorOperation):
286
286
  """
287
287
  Set the device for the current operator execution.
288
288
 
289
- - When the device is Ascend, input shape should be limited from [4, 6] to [8192, 4096].
289
+ When the device is Ascend, input shape should be limited from [4, 6] to [8192, 4096].
290
290
 
291
291
  Args:
292
292
  device_target (str, optional): The operator will be executed on this device. Currently supports
293
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
293
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
294
294
 
295
295
  Raises:
296
296
  TypeError: If `device_target` is not of type str.
297
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
297
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
298
298
 
299
299
  Supported Platforms:
300
300
  ``CPU`` ``Ascend``
@@ -357,14 +357,14 @@ class AdjustGamma(ImageTensorOperation, PyTensorOperation):
357
357
  Args:
358
358
  gamma (float): Non negative real number.
359
359
  The output image pixel value is exponentially related to the input image pixel value.
360
- gamma larger than 1 make the shadows darker,
361
- while gamma smaller than 1 make dark regions lighter.
360
+ gamma larger than ``1.0`` make the shadows darker,
361
+ while gamma smaller than ``1.0`` make dark regions lighter.
362
362
  gain (float, optional): The constant multiplier. Default: ``1.0``.
363
363
 
364
364
  Raises:
365
365
  TypeError: If `gain` is not of type float.
366
366
  TypeError: If `gamma` is not of type float.
367
- ValueError: If `gamma` is less than 0.
367
+ ValueError: If `gamma` is less than ``0.0``.
368
368
  RuntimeError: If given tensor shape is not <H, W> or <..., H, W, C>.
369
369
 
370
370
  Supported Platforms:
@@ -473,15 +473,15 @@ class AdjustHue(ImageTensorOperation, PyTensorOperation):
473
473
  """
474
474
  Set the device for the current operator execution.
475
475
 
476
- - When the device is Ascend, input shape should be limited from [4, 6] to [8192, 4096].
476
+ When the device is Ascend, input shape should be limited from [4, 6] to [8192, 4096].
477
477
 
478
478
  Args:
479
479
  device_target (str, optional): The operator will be executed on this device. Currently supports
480
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
480
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
481
481
 
482
482
  Raises:
483
483
  TypeError: If `device_target` is not of type str.
484
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
484
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
485
485
 
486
486
  Supported Platforms:
487
487
  ``CPU`` ``Ascend``
@@ -538,12 +538,12 @@ class AdjustSaturation(ImageTensorOperation, PyTensorOperation):
538
538
 
539
539
  Args:
540
540
  saturation_factor (float): How much to adjust the saturation, must be non negative.
541
- ``0`` gives a black image, ``1`` gives the original image
542
- while ``2`` increases the saturation by a factor of 2.
541
+ ``0.0`` gives a black image, ``1.0`` gives the original image
542
+ while ``2.0`` increases the saturation by a factor of 2.
543
543
 
544
544
  Raises:
545
545
  TypeError: If `saturation_factor` is not of type float.
546
- ValueError: If `saturation_factor` is less than 0.
546
+ ValueError: If `saturation_factor` is less than ``0.0``.
547
547
  RuntimeError: If shape of the input image is not <H, W, C>.
548
548
  RuntimeError: If channel of the input image is not 3.
549
549
 
@@ -586,15 +586,15 @@ class AdjustSaturation(ImageTensorOperation, PyTensorOperation):
586
586
  """
587
587
  Set the device for the current operator execution.
588
588
 
589
- - When the device is Ascend, input shape should be limited from [4, 6] to [8192, 4096].
589
+ When the device is Ascend, input shape should be limited from [4, 6] to [8192, 4096].
590
590
 
591
591
  Args:
592
592
  device_target (str, optional): The operator will be executed on this device. Currently supports
593
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
593
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
594
594
 
595
595
  Raises:
596
596
  TypeError: If `device_target` is not of type str.
597
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
597
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
598
598
 
599
599
  Supported Platforms:
600
600
  ``CPU`` ``Ascend``
@@ -651,12 +651,12 @@ class AdjustSharpness(ImageTensorOperation):
651
651
 
652
652
  Args:
653
653
  sharpness_factor (float): How much to adjust the sharpness, must be
654
- non negative. ``0`` gives a blurred image, ``1`` gives the
655
- original image while ``2`` increases the sharpness by a factor of 2.
654
+ non negative. ``0.0`` gives a blurred image, ``1.0`` gives the
655
+ original image while ``2.0`` increases the sharpness by a factor of 2.
656
656
 
657
657
  Raises:
658
658
  TypeError: If `sharpness_factor` is not of type float.
659
- ValueError: If `sharpness_factor` is less than 0.
659
+ ValueError: If `sharpness_factor` is less than ``0.0``.
660
660
  RuntimeError: If shape of the input image is not <H, W> or <H, W, C>.
661
661
 
662
662
  Supported Platforms:
@@ -700,16 +700,16 @@ class AdjustSharpness(ImageTensorOperation):
700
700
  """
701
701
  Set the device for the current operator execution.
702
702
 
703
- - When the device is Ascend, input type supports `uint8` or `float32` , input channel supports 1 and 3.
704
- The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
703
+ When the device is Ascend, input type supports `uint8` or `float32` , input channel supports 1 and 3.
704
+ The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
705
705
 
706
706
  Args:
707
707
  device_target (str, optional): The operator will be executed on this device. Currently supports
708
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
708
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
709
709
 
710
710
  Raises:
711
711
  TypeError: If `device_target` is not of type str.
712
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
712
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
713
713
 
714
714
  Supported Platforms:
715
715
  ``CPU`` ``Ascend``
@@ -754,13 +754,15 @@ class Affine(ImageTensorOperation):
754
754
  Supports Ascend hardware acceleration and can be enabled through the `.device("Ascend")` method.
755
755
 
756
756
  Args:
757
- degrees (float): Rotation angle in degrees between -180 and 180, clockwise direction.
757
+ degrees (float): Rotation angle in degrees between ``-180`` and ``180``, clockwise direction.
758
758
  translate (Sequence[float, float]): The horizontal and vertical translations, must be a sequence of size 2
759
- and value between -1 and 1.
759
+ and value between ``-1.0`` and ``1.0``.
760
760
  scale (float): Scaling factor, which must be positive.
761
- shear (Union[float, Sequence[float, float]]): Shear angle value in degrees between -180 to 180.
762
- If float is provided, shear along the x axis with this value, without shearing along the y axis;
763
- If Sequence[float, float] is provided, shear along the x axis and y axis with these two values separately.
761
+ shear (Union[float, Sequence[float, float]]): Shear angle value in degrees between ``-180`` to ``180``.
762
+
763
+ - If float is provided, shear along the x axis with this value, without shearing along the y axis;
764
+ - If Sequence[float, float] is provided, shear along the x axis and y axis with these two values separately.
765
+
764
766
  resample (Inter, optional): Image interpolation method defined by :class:`~.vision.Inter` .
765
767
  Default: ``Inter.NEAREST``.
766
768
  fill_value (Union[int, tuple[int, int, int]], optional): Optional `fill_value` to fill the area
@@ -832,15 +834,15 @@ class Affine(ImageTensorOperation):
832
834
  """
833
835
  Set the device for the current operator execution.
834
836
 
835
- - When the device is Ascend, input shape should be limited from [4, 6] to [32768, 32768].
837
+ When the device is Ascend, input shape should be limited from [4, 6] to [32768, 32768].
836
838
 
837
839
  Args:
838
840
  device_target (str, optional): The operator will be executed on this device. Currently supports
839
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
841
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
840
842
 
841
843
  Raises:
842
844
  TypeError: If `device_target` is not of type str.
843
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
845
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
844
846
 
845
847
  Supported Platforms:
846
848
  ``CPU`` ``Ascend``
@@ -904,9 +906,10 @@ class AutoAugment(ImageTensorOperation):
904
906
  interpolation (Inter, optional): Image interpolation method defined by :class:`~.vision.Inter` .
905
907
  Default: ``Inter.NEAREST``.
906
908
  fill_value (Union[int, tuple[int]], optional): Pixel fill value for the area outside the transformed image.
907
- It can be an int or a 3-tuple. If it is a 3-tuple, it is used to fill R, G, B channels respectively.
908
- If it is an integer, it is used for all RGB channels. The fill_value values must be in range [0, 255].
909
- Default: ``0``.
909
+ It can be an int or a 3-tuple. Default: ``0``.
910
+
911
+ - If it is a 3-tuple, it is used to fill R, G, B channels respectively.
912
+ - If it is an integer, it is used for all RGB channels. The fill_value values must be in range [0, 255].
910
913
 
911
914
  Raises:
912
915
  TypeError: If `policy` is not of type :class:`mindspore.dataset.vision.AutoAugmentPolicy` .
@@ -963,7 +966,8 @@ class AutoAugment(ImageTensorOperation):
963
966
 
964
967
  class AutoContrast(ImageTensorOperation, PyTensorOperation):
965
968
  """
966
- Apply automatic contrast on input image. This operation calculates histogram of image, reassign cutoff percent
969
+ Automatically adjusts the contrast on the input image. This operation calculates histogram of image,
970
+ reassign cutoff percent
967
971
  of the lightest pixels from histogram to 255, and reassign cutoff percent of the darkest pixels from histogram to 0.
968
972
 
969
973
  Supports Ascend hardware acceleration and can be enabled through the `.device("Ascend")` method.
@@ -977,7 +981,7 @@ class AutoContrast(ImageTensorOperation, PyTensorOperation):
977
981
  Raises:
978
982
  TypeError: If `cutoff` is not of type float.
979
983
  TypeError: If `ignore` is not of type int or sequence.
980
- ValueError: If `cutoff` is not in range [0, 50.0).
984
+ ValueError: If `cutoff` is not in range [0.0, 50.0).
981
985
  ValueError: If `ignore` is not in range [0, 255].
982
986
  RuntimeError: If given tensor shape is not <H, W> or <H, W, C>.
983
987
 
@@ -1026,17 +1030,17 @@ class AutoContrast(ImageTensorOperation, PyTensorOperation):
1026
1030
  """
1027
1031
  Set the device for the current operator execution.
1028
1032
 
1029
- - When the device is Ascend, input type supports `uint8` or `float32` , input channel supports 1 and 3.
1030
- If the data type is float32, the expected input value is in the range [0, 1].
1031
- The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
1033
+ When the device is Ascend, input type supports `uint8` or `float32` , input channel supports 1 and 3.
1034
+ If the data type is float32, the expected input value is in the range [0, 1].
1035
+ The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
1032
1036
 
1033
1037
  Args:
1034
1038
  device_target (str, optional): The operator will be executed on this device. Currently supports
1035
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
1039
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
1036
1040
 
1037
1041
  Raises:
1038
1042
  TypeError: If `device_target` is not of type str.
1039
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
1043
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
1040
1044
 
1041
1045
  Supported Platforms:
1042
1046
  ``CPU`` ``Ascend``
@@ -1343,16 +1347,16 @@ class ConvertColor(ImageTensorOperation):
1343
1347
  """
1344
1348
  Set the device for the current operator execution.
1345
1349
 
1346
- - When the device is Ascend, input type only supports `uint8` , input channel supports 1 and 3.
1347
- The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
1350
+ When the device is Ascend, input type only supports `uint8` , input channel supports 1 and 3.
1351
+ The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
1348
1352
 
1349
1353
  Args:
1350
1354
  device_target (str, optional): The operator will be executed on this device. Currently supports
1351
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
1355
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
1352
1356
 
1353
1357
  Raises:
1354
1358
  TypeError: If `device_target` is not of type str.
1355
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
1359
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
1356
1360
 
1357
1361
  Supported Platforms:
1358
1362
  ``CPU`` ``Ascend``
@@ -1399,8 +1403,10 @@ class Crop(ImageTensorOperation):
1399
1403
  coordinates(sequence): Coordinates of the upper left corner of the cropping image. Must be a sequence of two
1400
1404
  values, in the form of (top, left).
1401
1405
  size (Union[int, sequence]): The output size of the cropped image.
1402
- If size is an integer, a square crop of size (size, size) is returned.
1403
- If size is a sequence of length 2, it should be (height, width).
1406
+
1407
+ - If size is an integer, a square crop of size (size, size) is returned.
1408
+ - If size is a sequence of length 2, it should be (height, width).
1409
+
1404
1410
  The size value(s) must be larger than 0.
1405
1411
 
1406
1412
  Raises:
@@ -1454,15 +1460,15 @@ class Crop(ImageTensorOperation):
1454
1460
  """
1455
1461
  Set the device for the current operator execution.
1456
1462
 
1457
- - When the device is Ascend, input/output shape should be limited from [4, 6] to [32768, 32768].
1463
+ When the device is Ascend, input/output shape should be limited from [4, 6] to [32768, 32768].
1458
1464
 
1459
1465
  Args:
1460
1466
  device_target (str, optional): The operator will be executed on this device. Currently supports
1461
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
1467
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
1462
1468
 
1463
1469
  Raises:
1464
1470
  TypeError: If `device_target` is not of type str.
1465
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
1471
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
1466
1472
 
1467
1473
  Supported Platforms:
1468
1474
  ``CPU`` ``Ascend``
@@ -1735,13 +1741,15 @@ class Decode(ImageTensorOperation, PyTensorOperation):
1735
1741
  """
1736
1742
  Set the device for the current operator execution.
1737
1743
 
1744
+ When the device is Ascend, input type only supports `uint8`.
1745
+
1738
1746
  Args:
1739
1747
  device_target (str, optional): The operator will be executed on this device. Currently supports
1740
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
1748
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
1741
1749
 
1742
1750
  Raises:
1743
1751
  TypeError: If `device_target` is not of type str.
1744
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
1752
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
1745
1753
 
1746
1754
  Supported Platforms:
1747
1755
  ``CPU`` ``Ascend``
@@ -1918,16 +1926,16 @@ class Equalize(ImageTensorOperation, PyTensorOperation):
1918
1926
  """
1919
1927
  Set the device for the current operator execution.
1920
1928
 
1921
- - When the device is Ascend, input type only supports `uint8` , input channel supports 1 and 3.
1922
- The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
1929
+ When the device is Ascend, input type only supports `uint8` , input channel supports 1 and 3.
1930
+ The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
1923
1931
 
1924
1932
  Args:
1925
1933
  device_target (str, optional): The operator will be executed on this device. Currently supports
1926
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
1934
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
1927
1935
 
1928
1936
  Raises:
1929
1937
  TypeError: If `device_target` is not of type str.
1930
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
1938
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
1931
1939
 
1932
1940
  Supported Platforms:
1933
1941
  ``CPU`` ``Ascend``
@@ -1979,7 +1987,7 @@ class Equalize(ImageTensorOperation, PyTensorOperation):
1979
1987
 
1980
1988
  class Erase(ImageTensorOperation):
1981
1989
  """
1982
- Erase the input image with given value.
1990
+ Erase a specified area of the input image and fills it with the specified pixel value `value`.
1983
1991
 
1984
1992
  Supports Ascend hardware acceleration and can be enabled through the `.device("Ascend")` method.
1985
1993
 
@@ -1989,8 +1997,11 @@ class Erase(ImageTensorOperation):
1989
1997
  height (int): Height of erased region.
1990
1998
  width (int): Width of erased region.
1991
1999
  value (Union[float, Sequence[float, float, float]], optional): Pixel value used to pad the erased area.
1992
- Default: ``0``. If float is provided, it will be used for all RGB channels.
1993
- If Sequence[float, float, float] is provided, it will be used for R, G, B channels respectively.
2000
+ Default: ``0.0``.
2001
+
2002
+ - If float is provided, it will be used for all RGB channels.
2003
+ - If Sequence[float, float, float] is provided, it will be used for R, G, B channels respectively.
2004
+
1994
2005
  inplace (bool, optional): Whether to apply erasing inplace. Default: ``False``.
1995
2006
 
1996
2007
  Raises:
@@ -2037,7 +2048,7 @@ class Erase(ImageTensorOperation):
2037
2048
  """
2038
2049
 
2039
2050
  @check_erase
2040
- def __init__(self, top, left, height, width, value=0, inplace=False):
2051
+ def __init__(self, top, left, height, width, value=0.0, inplace=False):
2041
2052
  super().__init__()
2042
2053
  self.top = top
2043
2054
  self.left = left
@@ -2053,17 +2064,17 @@ class Erase(ImageTensorOperation):
2053
2064
  """
2054
2065
  Set the device for the current operator execution.
2055
2066
 
2056
- - When the device is Ascend, input type supports `uint8` or `float32` , input channel supports 1 and 3.
2057
- The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
2058
- The inplace parameter is not supported.
2067
+ When the device is Ascend, input type supports `uint8` or `float32` , input channel supports 1 and 3.
2068
+ The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
2069
+ The inplace parameter is not supported.
2059
2070
 
2060
2071
  Args:
2061
2072
  device_target (str, optional): The operator will be executed on this device. Currently supports
2062
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
2073
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
2063
2074
 
2064
2075
  Raises:
2065
2076
  TypeError: If `device_target` is not of type str.
2066
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
2077
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
2067
2078
 
2068
2079
  Supported Platforms:
2069
2080
  ``CPU`` ``Ascend``
@@ -2203,15 +2214,19 @@ class GaussianBlur(ImageTensorOperation):
2203
2214
 
2204
2215
  Args:
2205
2216
  kernel_size (Union[int, Sequence[int, int]]): The size of the Gaussian kernel. Must be positive and odd.
2206
- If the input type is int, the value will be used as both the width and height of the Gaussian kernel.
2207
- If the input type is Sequence[int, int], the two elements will be used as the width and height of the
2208
- Gaussian kernel respectively.
2217
+
2218
+ - If the input type is int, the value will be used as both the width and height of the Gaussian kernel.
2219
+ - If the input type is Sequence[int, int], the two elements will be used as the width and height of the
2220
+ Gaussian kernel respectively.
2221
+
2209
2222
  sigma (Union[float, Sequence[float, float]], optional): The standard deviation of the Gaussian kernel.
2210
2223
  Must be positive.
2211
- If the input type is float, the value will be used as the standard deviation of both the width and
2212
- height of the Gaussian kernel.
2213
- If the input type is Sequence[float, float], the two elements will be used as the standard deviation
2214
- of the width and height of the Gaussian kernel respectively.
2224
+
2225
+ - If the input type is float, the value will be used as the standard deviation of both the width and
2226
+ height of the Gaussian kernel.
2227
+ - If the input type is Sequence[float, float], the two elements will be used as the standard deviation
2228
+ of the width and height of the Gaussian kernel respectively.
2229
+
2215
2230
  Default: ``None`` , the standard deviation of the Gaussian kernel will be obtained by the
2216
2231
  formula :math:`((kernel\_size - 1) * 0.5 - 1) * 0.3 + 0.8` .
2217
2232
 
@@ -2269,16 +2284,16 @@ class GaussianBlur(ImageTensorOperation):
2269
2284
  """
2270
2285
  Set the device for the current operator execution.
2271
2286
 
2272
- - When the device is Ascend, the parameter `kernel_size` only supports values 1, 3, and 5.
2273
- input shape should be limited from [4, 6] to [8192, 4096].
2287
+ When the device is Ascend, the parameter `kernel_size` only supports values 1, 3, and 5.
2288
+ input shape should be limited from [4, 6] to [8192, 4096].
2274
2289
 
2275
2290
  Args:
2276
2291
  device_target (str, optional): The operator will be executed on this device. Currently supports
2277
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
2292
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
2278
2293
 
2279
2294
  Raises:
2280
2295
  TypeError: If `device_target` is not of type str.
2281
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
2296
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
2282
2297
 
2283
2298
  Supported Platforms:
2284
2299
  ``CPU`` ``Ascend``
@@ -2456,17 +2471,17 @@ class HorizontalFlip(ImageTensorOperation):
2456
2471
  """
2457
2472
  Set the device for the current operator execution.
2458
2473
 
2459
- - When the device is Ascend, input type supports `uint8` and `float32`,
2460
- input channel supports 1 and 3. The input data has a height limit of [4, 8192]
2461
- and a width limit of [6, 4096].
2474
+ When the device is Ascend, input type supports `uint8` and `float32`,
2475
+ input channel supports 1 and 3. The input data has a height limit of [4, 8192]
2476
+ and a width limit of [6, 4096].
2462
2477
 
2463
2478
  Args:
2464
2479
  device_target (str, optional): The operator will be executed on this device. Currently supports
2465
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
2480
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
2466
2481
 
2467
2482
  Raises:
2468
2483
  TypeError: If `device_target` is not of type str.
2469
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
2484
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
2470
2485
 
2471
2486
  Supported Platforms:
2472
2487
  ``CPU`` ``Ascend``
@@ -2670,9 +2685,8 @@ class Invert(ImageTensorOperation, PyTensorOperation):
2670
2685
  """
2671
2686
  Set the device for the current operator execution.
2672
2687
 
2673
- - When the device is CPU, input type only support `uint8` , input channel support 1/2/3.
2674
- - When the device is Ascend, input type supports `uint8`/`float32`, input channel supports 1/3.
2675
- input shape should be limited from [4, 6] to [8192, 4096].
2688
+ When the device is Ascend, input type supports `uint8`/`float32`, input channel supports 1/3.
2689
+ input shape should be limited from [4, 6] to [8192, 4096].
2676
2690
 
2677
2691
  Args:
2678
2692
  device_target (str, optional): The operator will be executed on this device. Currently supports
@@ -3057,11 +3071,11 @@ class Normalize(ImageTensorOperation):
3057
3071
 
3058
3072
  Args:
3059
3073
  device_target (str, optional): The operator will be executed on this device. Currently supports
3060
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
3074
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
3061
3075
 
3062
3076
  Raises:
3063
3077
  TypeError: If `device_target` is not of type str.
3064
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
3078
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
3065
3079
 
3066
3080
  Supported Platforms:
3067
3081
  ``CPU`` ``Ascend``
@@ -3113,7 +3127,7 @@ class NormalizePad(ImageTensorOperation):
3113
3127
  The standard deviation values must be in range (0.0, 255.0].
3114
3128
  dtype (str, optional): Set the output data type of normalized image. Default: ``"float32"``.
3115
3129
  is_hwc (bool, optional): Specify the format of input image.
3116
- ``True`` - HW(C) format, ``False`` - CHW format. Default: ``True``.
3130
+ ``True`` - HWC format, ``False`` - CHW format. Default: ``True``.
3117
3131
 
3118
3132
  Raises:
3119
3133
  TypeError: If `mean` is not of type sequence.
@@ -3175,14 +3189,20 @@ class Pad(ImageTensorOperation, PyTensorOperation):
3175
3189
  Args:
3176
3190
  padding (Union[int, Sequence[int, int], Sequence[int, int, int, int]]): The number of pixels
3177
3191
  to pad each border of the image.
3178
- If a single number is provided, it pads all borders with this value.
3179
- If a tuple or lists of 2 values are provided, it pads the (left and right)
3180
- with the first value and (top and bottom) with the second value.
3181
- If 4 values are provided as a list or tuple, it pads the left, top, right and bottom respectively.
3192
+
3193
+ - If a single number is provided, it pads all borders with this value.
3194
+ - If a tuple or lists of 2 values are provided, it pads the (left and right)
3195
+ with the first value and (top and bottom) with the second value.
3196
+ - If 4 values are provided as a list or tuple, it pads the left, top, right and bottom respectively.
3197
+
3182
3198
  The pad values must be non-negative.
3199
+
3183
3200
  fill_value (Union[int, tuple[int]], optional): The pixel intensity of the borders, only valid for
3184
- `padding_mode` ``Border.CONSTANT``. If it is a 3-tuple, it is used to fill R, G, B channels respectively.
3185
- If it is an integer, it is used for all RGB channels.
3201
+ `padding_mode` ``Border.CONSTANT``.
3202
+
3203
+ - If it is a 3-tuple, it is used to fill R, G, B channels respectively.
3204
+ - If it is an integer, it is used for all RGB channels.
3205
+
3186
3206
  The fill_value values must be in range [0, 255]. Default: ``0``.
3187
3207
  padding_mode (Border, optional): The method of padding. Default: ``Border.CONSTANT``. Can be
3188
3208
  ``Border.CONSTANT``, ``Border.EDGE``, ``Border.REFLECT``, ``Border.SYMMETRIC``.
@@ -3250,15 +3270,15 @@ class Pad(ImageTensorOperation, PyTensorOperation):
3250
3270
  """
3251
3271
  Set the device for the current operator execution.
3252
3272
 
3253
- - When the device is Ascend, input/output shape should be limited from [4, 6] to [32768, 32768].
3273
+ When the device is Ascend, input/output shape should be limited from [4, 6] to [32768, 32768].
3254
3274
 
3255
3275
  Args:
3256
3276
  device_target (str, optional): The operator will be executed on this device. Currently supports
3257
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
3277
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
3258
3278
 
3259
3279
  Raises:
3260
3280
  TypeError: If `device_target` is not of type str.
3261
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
3281
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
3262
3282
 
3263
3283
  Supported Platforms:
3264
3284
  ``CPU`` ``Ascend``
@@ -3403,6 +3423,9 @@ class Perspective(ImageTensorOperation, PyTensorOperation):
3403
3423
  interpolation (Inter, optional): Image interpolation method defined by :class:`~.vision.Inter` .
3404
3424
  Default: ``Inter.BILINEAR``.
3405
3425
 
3426
+ - Ascend Mode: Only ``Inter.LINEAR`` and ``Inter.NEAREST`` interpolation methods are supported
3427
+ when the device is set to Ascend by `.device("Ascend")` .
3428
+
3406
3429
  Raises:
3407
3430
  TypeError: If `start_points` is not of type Sequence[Sequence[int, int]].
3408
3431
  TypeError: If `end_points` is not of type Sequence[Sequence[int, int]].
@@ -3461,17 +3484,17 @@ class Perspective(ImageTensorOperation, PyTensorOperation):
3461
3484
  """
3462
3485
  Set the device for the current operator execution.
3463
3486
 
3464
- - When the device is Ascend, input type supports `uint8` and `float32`,
3465
- input channel supports 1 and 3. The input data has a height limit of [6, 8192]
3466
- and a width limit of [10, 4096].
3487
+ When the device is Ascend, input type supports `uint8` and `float32`,
3488
+ input channel supports 1 and 3. The input data has a height limit of [6, 8192]
3489
+ and a width limit of [10, 4096].
3467
3490
 
3468
3491
  Args:
3469
3492
  device_target (str, optional): The operator will be executed on this device. Currently supports
3470
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
3493
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
3471
3494
 
3472
3495
  Raises:
3473
3496
  TypeError: If `device_target` is not of type str.
3474
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
3497
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
3475
3498
 
3476
3499
  Supported Platforms:
3477
3500
  ``CPU`` ``Ascend``
@@ -3585,16 +3608,16 @@ class Posterize(ImageTensorOperation):
3585
3608
  """
3586
3609
  Set the device for the current operator execution.
3587
3610
 
3588
- - When the device is Ascend, input type supports `uint8`/`float32`, input channel supports 1 and 3.
3589
- The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
3611
+ When the device is Ascend, input type supports `uint8`/`float32`, input channel supports 1 and 3.
3612
+ The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
3590
3613
 
3591
3614
  Args:
3592
3615
  device_target (str, optional): The operator will be executed on this device. Currently supports
3593
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
3616
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
3594
3617
 
3595
3618
  Raises:
3596
3619
  TypeError: If `device_target` is not of type str.
3597
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
3620
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
3598
3621
 
3599
3622
  Supported Platforms:
3600
3623
  ``CPU`` ``Ascend``
@@ -3779,26 +3802,30 @@ class RandomAffine(ImageTensorOperation, PyTensorOperation):
3779
3802
  If `degrees` is a number, the range will be (-degrees, degrees).
3780
3803
  If `degrees` is a sequence, it should be (min, max).
3781
3804
  translate (sequence, optional): Sequence (tx_min, tx_max, ty_min, ty_max) of minimum/maximum translation in
3782
- x(horizontal) and y(vertical) directions, range [-1.0, 1.0]. Default: ``None``.
3805
+ x(horizontal) and y(vertical) directions, range [-1.0, 1.0].
3806
+ Default: ``None``, which means no translation is applied.
3783
3807
  The horizontal and vertical shift is selected randomly from the range:
3784
3808
  (tx_min*width, tx_max*width) and (ty_min*height, ty_max*height), respectively.
3785
- If a tuple or list of size 2, then a translate parallel to the X axis in the range of
3786
- (translate[0], translate[1]) is applied.
3787
- If a tuple or list of size 4, then a translate parallel to the X axis in the range of
3788
- (translate[0], translate[1]) and a translate parallel to the Y axis in the range of
3789
- (translate[2], translate[3]) are applied.
3790
- If ``None``, no translation is applied.
3809
+
3810
+ - If a tuple or list of size 2, then a translate parallel to the X axis in the range of
3811
+ (translate[0], translate[1]) is applied.
3812
+ - If a tuple or list of size 4, then a translate parallel to the X axis in the range of
3813
+ (translate[0], translate[1]) and a translate parallel to the Y axis in the range of
3814
+ (translate[2], translate[3]) are applied.
3815
+
3791
3816
  scale (sequence, optional): Scaling factor interval, which must be non negative.
3792
3817
  Default: ``None``, original scale is used.
3793
3818
  shear (Union[float, Sequence[float, float], Sequence[float, float, float, float]], optional):
3794
- Range of shear factor to select from.
3795
- If float is provided, a shearing parallel to X axis with a factor selected from
3796
- ( `-shear` , `shear` ) will be applied.
3797
- If Sequence[float, float] is provided, a shearing parallel to X axis with a factor selected
3798
- from ( `shear` [0], `shear` [1]) will be applied.
3799
- If Sequence[float, float, float, float] is provided, a shearing parallel to X axis with a factor selected
3800
- from ( `shear` [0], `shear` [1]) and a shearing parallel to Y axis with a factor selected from
3801
- ( `shear` [2], `shear` [3]) will be applied. Default: ``None``, means no shearing.
3819
+ Range of shear factor to select from. Default: ``None``, means no shearing.
3820
+
3821
+ - If float is provided, a shearing parallel to X axis with a factor selected from
3822
+ ( `-shear` , `shear` ) will be applied.
3823
+ - If Sequence[float, float] is provided, a shearing parallel to X axis with a factor selected
3824
+ from ( `shear` [0], `shear` [1]) will be applied.
3825
+ - If Sequence[float, float, float, float] is provided, a shearing parallel to X axis with a factor selected
3826
+ from ( `shear` [0], `shear` [1]) and a shearing parallel to Y axis with a factor selected from
3827
+ ( `shear` [2], `shear` [3]) will be applied.
3828
+
3802
3829
  resample (Inter, optional): Image interpolation method defined by :class:`~.vision.Inter` .
3803
3830
  Default: ``Inter.NEAREST``.
3804
3831
  fill_value (Union[int, tuple[int]], optional): Optional fill_value to fill the area outside the transform
@@ -4172,22 +4199,28 @@ class RandomCrop(ImageTensorOperation, PyTensorOperation):
4172
4199
 
4173
4200
  Args:
4174
4201
  size (Union[int, Sequence[int]]): The output size of the cropped image. The size value(s) must be positive.
4175
- If size is an integer, a square crop of size (size, size) is returned.
4176
- If size is a sequence of length 2, an image of size (height, width) will be cropped.
4202
+
4203
+ - If size is an integer, a square crop of size (size, size) is returned.
4204
+ - If size is a sequence of length 2, an image of size (height, width) will be cropped.
4205
+
4177
4206
  padding (Union[int, Sequence[int]], optional): The number of pixels to pad each border of the image.
4178
4207
  The padding value(s) must be non-negative. Default: ``None``.
4179
- If `padding` is not ``None``, pad image first with padding values.
4180
- If a single number is provided, pad all borders with this value.
4181
- If a tuple or lists of 2 values are provided, pad the (left and right)
4182
- with the first value and (top and bottom) with the second value.
4183
- If 4 values are provided as a list or tuple,
4184
- pad the left, top, right and bottom respectively.
4208
+
4209
+ - If `padding` is not ``None``, pad image first with padding values.
4210
+ - If a single number is provided, pad all borders with this value.
4211
+ - If a tuple or lists of 2 values are provided, pad the (left and right)
4212
+ with the first value and (top and bottom) with the second value.
4213
+ - If 4 values are provided as a list or tuple,
4214
+ pad the left, top, right and bottom respectively.
4215
+
4185
4216
  pad_if_needed (bool, optional): Pad the image if either side is smaller than
4186
4217
  the given output size. Default: ``False``.
4187
4218
  fill_value (Union[int, tuple[int]], optional): The pixel intensity of the borders, only valid for
4188
- padding_mode Border.CONSTANT. If it is a 3-tuple, it is used to fill R, G, B channels respectively.
4189
- If it is an integer, it is used for all RGB channels.
4190
- The fill_value values must be in range [0, 255]. Default: ``0``.
4219
+ padding_mode Border.CONSTANT. The fill_value values must be in range [0, 255]. Default: ``0``.
4220
+
4221
+ - If it is a 3-tuple, it is used to fill R, G, B channels respectively.
4222
+ - If it is an integer, it is used for all RGB channels.
4223
+
4191
4224
  padding_mode (Border, optional): The method of padding. Default: ``Border.CONSTANT``. It can be any of
4192
4225
  ``Border.CONSTANT``, ``Border.EDGE``, ``Border.REFLECT``, ``Border.SYMMETRIC``.
4193
4226
 
@@ -5352,7 +5385,7 @@ class RandomResizedCropWithBBox(ImageTensorOperation):
5352
5385
 
5353
5386
  class RandomResize(ImageTensorOperation):
5354
5387
  """
5355
- Resize the input image using :class:`~.vision.Inter` , a randomly selected interpolation mode.
5388
+ Resize the input image using a randomly selected :class:`mindspore.dataset.vision.Inter` interpolation method.
5356
5389
 
5357
5390
  Args:
5358
5391
  size (Union[int, Sequence[int]]): The output size of the resized image. The size value(s) must be positive.
@@ -5996,9 +6029,18 @@ class Resize(ImageTensorOperation, PyTensorOperation):
5996
6029
  If size is an integer, the smaller edge of the image will be resized to this value with
5997
6030
  the same image aspect ratio.
5998
6031
  If size is a sequence of length 2, it should be (height, width).
6032
+
6033
+ - CPU mode: If the execution device is CPU by ``.device("CPU")`` , the range of size is [1, 16777216].
6034
+
6035
+ - Ascend mode: set the executing device to Ascend by ``.device("Ascend")`` , the value of size will be in
6036
+ the range of [6, 32768].
6037
+
5999
6038
  interpolation (Inter, optional): Image interpolation method defined by :class:`~.vision.Inter` .
6000
6039
  Default: ``Inter.LINEAR``.
6001
6040
 
6041
+ - Ascend mode: ``Inter.ANTIALIAS`` , ``Inter.AREA`` , ``Inter.PILCUBIC`` interpolation methods are not
6042
+ supported when the execution device is set to Ascend by ``.device("Ascend")`` .
6043
+
6002
6044
  Raises:
6003
6045
  TypeError: If `size` is not of type int or Sequence[int].
6004
6046
  TypeError: If `interpolation` is not of type :class:`~.vision.Inter` .
@@ -6055,15 +6097,15 @@ class Resize(ImageTensorOperation, PyTensorOperation):
6055
6097
  """
6056
6098
  Set the device for the current operator execution.
6057
6099
 
6058
- - When the device is Ascend, input/output shape should be limited from [4, 6] to [32768, 32768].
6100
+ When the device is Ascend, input/output shape should be limited from [4, 6] to [32768, 32768].
6059
6101
 
6060
6102
  Args:
6061
6103
  device_target (str, optional): The operator will be executed on this device. Currently supports
6062
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
6104
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
6063
6105
 
6064
6106
  Raises:
6065
6107
  TypeError: If `device_target` is not of type str.
6066
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
6108
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
6067
6109
 
6068
6110
  Supported Platforms:
6069
6111
  ``CPU`` ``Ascend``
@@ -6132,9 +6174,11 @@ class ResizedCrop(ImageTensorOperation):
6132
6174
  height (int): Height of the crop region.
6133
6175
  width (int): Width of the cropp region.
6134
6176
  size (Union[int, Sequence[int, int]]): The size of the output image.
6135
- If int is provided, the smaller edge of the image will be resized to this value,
6136
- keeping the image aspect ratio the same.
6137
- If Sequence[int, int] is provided, it should be (height, width).
6177
+
6178
+ - If int is provided, the smaller edge of the image will be resized to this value,
6179
+ keeping the image aspect ratio the same.
6180
+ - If Sequence[int, int] is provided, it should be (height, width).
6181
+
6138
6182
  interpolation (Inter, optional): Image interpolation method defined by :class:`~.vision.Inter` .
6139
6183
  Default: ``Inter.BILINEAR``.
6140
6184
 
@@ -6201,17 +6245,17 @@ class ResizedCrop(ImageTensorOperation):
6201
6245
  """
6202
6246
  Set the device for the current operator execution.
6203
6247
 
6204
- - When the device is Ascend, input type supports `uint8` and `float32`,
6205
- input channel supports 1 and 3. The input data has a height limit of [4, 32768]
6206
- and a width limit of [6, 32768].
6248
+ When the device is Ascend, input type supports `uint8` and `float32`,
6249
+ input channel supports 1 and 3. The input data has a height limit of [4, 32768]
6250
+ and a width limit of [6, 32768].
6207
6251
 
6208
6252
  Args:
6209
6253
  device_target (str, optional): The operator will be executed on this device. Currently supports
6210
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
6254
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
6211
6255
 
6212
6256
  Raises:
6213
6257
  TypeError: If `device_target` is not of type str.
6214
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
6258
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
6215
6259
 
6216
6260
  Supported Platforms:
6217
6261
  ``CPU`` ``Ascend``
@@ -6478,11 +6522,11 @@ class Rotate(ImageTensorOperation):
6478
6522
 
6479
6523
  Args:
6480
6524
  device_target (str, optional): The operator will be executed on this device. Currently supports
6481
- ``CPU`` and ``Ascend`` . Default: ``CPU`` .
6525
+ ``"CPU"`` and ``"Ascend"`` . Default: ``"CPU"`` .
6482
6526
 
6483
6527
  Raises:
6484
6528
  TypeError: If `device_target` is not of type str.
6485
- ValueError: If `device_target` is not within the valid set of ['CPU', 'Ascend'].
6529
+ ValueError: If `device_target` is not within the valid set of ["CPU", "Ascend"].
6486
6530
 
6487
6531
  Supported Platforms:
6488
6532
  ``CPU`` ``Ascend``
@@ -6660,8 +6704,8 @@ class Solarize(ImageTensorOperation):
6660
6704
  """
6661
6705
  Set the device for the current operator execution.
6662
6706
 
6663
- - When the device is Ascend, input type only supports `uint8` , input channel supports 1 and 3.
6664
- The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
6707
+ When the device is Ascend, input type only supports `uint8` , input channel supports 1 and 3.
6708
+ The input data has a height limit of [4, 8192] and a width limit of [6, 4096].
6665
6709
 
6666
6710
  Args:
6667
6711
  device_target (str, optional): The operator will be executed on this device. Currently supports
@@ -7238,9 +7282,9 @@ class VerticalFlip(ImageTensorOperation):
7238
7282
  """
7239
7283
  Set the device for the current operator execution.
7240
7284
 
7241
- - When the device is Ascend, input type supports `uint8` and `float32`,
7242
- input channel supports 1 and 3. The input data has a height limit of [4, 8192]
7243
- and a width limit of [6, 4096].
7285
+ When the device is Ascend, input type supports `uint8` and `float32`,
7286
+ input channel supports 1 and 3. The input data has a height limit of [4, 8192]
7287
+ and a width limit of [6, 4096].
7244
7288
 
7245
7289
  Args:
7246
7290
  device_target (str, optional): The operator will be executed on this device. Currently supports