mindspore 1.10.0__cp38-cp38-win_amd64.whl → 2.0.0rc1__cp38-cp38-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.
- mindspore/.commit_id +1 -1
- mindspore/ConcurrencyCheck.dll +0 -0
- mindspore/CppBuildInsights.dll +0 -0
- mindspore/CppCoreCheck.dll +0 -0
- mindspore/EnumIndex.dll +0 -0
- mindspore/EspXEngine.dll +0 -0
- mindspore/HResultCheck.dll +0 -0
- mindspore/KernelTraceControl.dll +0 -0
- mindspore/LocalESPC.dll +0 -0
- mindspore/Microsoft.Diagnostics.Tracing.EventSource.dll +0 -0
- mindspore/Microsoft.VisualStudio.RemoteControl.dll +0 -0
- mindspore/Microsoft.VisualStudio.Telemetry.dll +0 -0
- mindspore/Microsoft.VisualStudio.Utilities.Internal.dll +0 -0
- mindspore/Newtonsoft.Json.dll +0 -0
- mindspore/System.Runtime.CompilerServices.Unsafe.dll +0 -0
- mindspore/VariantClear.dll +0 -0
- mindspore/__init__.py +9 -4
- mindspore/_c_dataengine.cp38-win_amd64.pyd +0 -0
- mindspore/_c_expression.cp38-win_amd64.pyd +0 -0
- mindspore/_c_mindrecord.cp38-win_amd64.pyd +0 -0
- mindspore/_check_jit_forbidden_api.py +102 -0
- mindspore/_checkparam.py +1066 -1001
- mindspore/_extends/builtin_operations.py +32 -4
- mindspore/_extends/graph_kernel/model/graph_split.py +66 -222
- mindspore/_extends/parallel_compile/akg_compiler/akg_process.py +12 -9
- mindspore/_extends/parallel_compile/akg_compiler/build_tbe_kernel.py +119 -26
- mindspore/_extends/parallel_compile/akg_compiler/tbe_topi.py +50 -50
- mindspore/_extends/parallel_compile/akg_compiler/util.py +9 -6
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_adapter.py +4 -25
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_helper.py +9 -4
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_job_manager.py +1 -27
- mindspore/_extends/parse/__init__.py +5 -3
- mindspore/_extends/parse/namespace.py +17 -2
- mindspore/_extends/parse/parser.py +193 -34
- mindspore/_extends/parse/resources.py +7 -8
- mindspore/_extends/parse/standard_method.py +1780 -435
- mindspore/_extends/parse/trope.py +3 -1
- mindspore/amp.py +53 -58
- mindspore/atlprov.dll +0 -0
- mindspore/boost/adasum.py +3 -2
- mindspore/boost/boost.py +2 -2
- mindspore/boost/boost_cell_wrapper.py +46 -26
- mindspore/boost/dim_reduce.py +6 -5
- mindspore/boost/grad_accumulation.py +2 -1
- mindspore/boost/group_loss_scale_manager.py +1 -1
- mindspore/c1.dll +0 -0
- mindspore/c1xx.dll +0 -0
- mindspore/c2.dll +0 -0
- mindspore/cfgpersist.dll +0 -0
- mindspore/clang_rt.asan_dbg_dynamic-x86_64.dll +0 -0
- mindspore/clang_rt.asan_dynamic-x86_64.dll +0 -0
- mindspore/common/__init__.py +11 -10
- mindspore/common/_decorator.py +2 -0
- mindspore/common/_register_for_adapter.py +55 -0
- mindspore/common/_stub_tensor.py +201 -0
- mindspore/common/_utils.py +57 -0
- mindspore/common/api.py +582 -297
- mindspore/common/dtype.py +66 -18
- mindspore/common/dump.py +2 -2
- mindspore/common/initializer.py +38 -1
- mindspore/common/jit_config.py +25 -13
- mindspore/common/mutable.py +53 -24
- mindspore/common/parameter.py +60 -37
- mindspore/common/seed.py +8 -24
- mindspore/common/sparse_tensor.py +927 -0
- mindspore/common/tensor.py +1627 -3900
- mindspore/communication/__init__.py +10 -5
- mindspore/communication/_comm_helper.py +78 -214
- mindspore/communication/_hccl_management.py +2 -1
- mindspore/communication/management.py +136 -47
- mindspore/config/op_info.config +501 -1008
- mindspore/context.py +291 -56
- mindspore/d3dcompiler_47.dll +0 -0
- mindspore/dataset/__init__.py +12 -8
- mindspore/dataset/audio/__init__.py +9 -9
- mindspore/dataset/audio/transforms.py +1090 -228
- mindspore/dataset/audio/utils.py +87 -39
- mindspore/dataset/audio/validators.py +223 -1
- mindspore/dataset/callback/ds_callback.py +17 -15
- mindspore/dataset/core/config.py +246 -17
- mindspore/dataset/core/py_util_helpers.py +4 -3
- mindspore/dataset/core/validator_helpers.py +10 -10
- mindspore/{parallel/nn/layers.py → dataset/debug/__init__.py} +7 -8
- mindspore/dataset/debug/debug_hook.py +65 -0
- mindspore/dataset/debug/pre_defined_hook.py +67 -0
- mindspore/dataset/engine/__init__.py +7 -3
- mindspore/dataset/engine/cache_client.py +9 -9
- mindspore/dataset/engine/datasets.py +648 -477
- mindspore/dataset/engine/datasets_audio.py +165 -167
- mindspore/dataset/engine/datasets_standard_format.py +93 -67
- mindspore/dataset/engine/datasets_text.py +492 -342
- mindspore/dataset/engine/datasets_user_defined.py +85 -50
- mindspore/dataset/engine/datasets_vision.py +1224 -699
- mindspore/dataset/engine/graphdata.py +134 -69
- mindspore/dataset/engine/iterators.py +50 -9
- mindspore/dataset/engine/offload.py +52 -31
- mindspore/dataset/engine/samplers.py +27 -24
- mindspore/dataset/engine/serializer_deserializer.py +14 -15
- mindspore/dataset/engine/validators.py +213 -52
- mindspore/dataset/text/__init__.py +10 -8
- mindspore/dataset/text/transforms.py +152 -57
- mindspore/dataset/text/utils.py +98 -49
- mindspore/dataset/text/validators.py +25 -0
- mindspore/dataset/transforms/__init__.py +4 -2
- mindspore/dataset/transforms/c_transforms.py +11 -13
- mindspore/dataset/transforms/py_transforms.py +2 -2
- mindspore/dataset/transforms/py_transforms_util.py +10 -0
- mindspore/dataset/transforms/transforms.py +13 -15
- mindspore/dataset/transforms/validators.py +7 -7
- mindspore/dataset/utils/__init__.py +2 -1
- mindspore/dataset/utils/browse_dataset.py +13 -13
- mindspore/dataset/utils/line_reader.py +121 -0
- mindspore/dataset/vision/__init__.py +8 -7
- mindspore/dataset/vision/c_transforms.py +125 -126
- mindspore/dataset/vision/py_transforms.py +37 -37
- mindspore/dataset/vision/py_transforms_util.py +23 -20
- mindspore/dataset/vision/transforms.py +316 -315
- mindspore/dataset/vision/utils.py +313 -17
- mindspore/dataset/vision/validators.py +6 -6
- mindspore/default_config.py +0 -1
- mindspore/dpcmi.dll +0 -0
- mindspore/{compression → experimental}/__init__.py +6 -5
- mindspore/experimental/map_parameter.py +275 -0
- mindspore/include/OWNERS +0 -1
- mindspore/include/api/callback/callback.h +9 -13
- mindspore/include/api/callback/ckpt_saver.h +2 -2
- mindspore/include/api/callback/loss_monitor.h +2 -2
- mindspore/include/api/callback/lr_scheduler.h +5 -5
- mindspore/include/api/callback/time_monitor.h +2 -2
- mindspore/include/api/callback/train_accuracy.h +4 -6
- mindspore/include/api/cfg.h +19 -6
- mindspore/include/api/context.h +70 -9
- mindspore/include/api/delegate.h +8 -1
- mindspore/include/api/dual_abi_helper.h +8 -24
- mindspore/include/api/metrics/accuracy.h +2 -2
- mindspore/include/api/metrics/metrics.h +4 -3
- mindspore/include/api/model.h +9 -4
- mindspore/include/api/model_group.h +68 -0
- mindspore/include/api/model_parallel_runner.h +17 -17
- mindspore/include/api/net.h +12 -11
- mindspore/include/api/serialization.h +20 -4
- mindspore/include/api/status.h +7 -1
- mindspore/include/api/types.h +25 -21
- mindspore/include/api/visible.h +4 -0
- mindspore/include/c_api/model_c.h +5 -0
- mindspore/include/c_api/status_c.h +1 -1
- mindspore/include/dataset/config.h +1 -1
- mindspore/include/dataset/constants.h +14 -0
- mindspore/include/dataset/text.h +59 -0
- mindspore/include/dataset/vision.h +56 -117
- mindspore/include/dataset/vision_lite.h +102 -0
- mindspore/jpeg62.dll +0 -0
- mindspore/log.py +28 -28
- mindspore/mindrecord/common/exceptions.py +2 -4
- mindspore/mindrecord/filereader.py +19 -1
- mindspore/mindrecord/filewriter.py +250 -88
- mindspore/mindrecord/mindpage.py +13 -13
- mindspore/mindrecord/shardheader.py +15 -15
- mindspore/mindrecord/shardreader.py +9 -0
- mindspore/mindrecord/shardwriter.py +29 -29
- mindspore/mindrecord/tools/cifar100_to_mr.py +9 -9
- mindspore/mindrecord/tools/cifar10_to_mr.py +9 -9
- mindspore/mindrecord/tools/csv_to_mr.py +4 -4
- mindspore/mindrecord/tools/imagenet_to_mr.py +70 -65
- mindspore/mindrecord/tools/mnist_to_mr.py +41 -41
- mindspore/mindrecord/tools/tfrecord_to_mr.py +6 -6
- mindspore/{libmindspore_backend.dll → mindspore_backend.dll} +0 -0
- mindspore/mindspore_common.dll +0 -0
- mindspore/mindspore_core.dll +0 -0
- mindspore/mindspore_glog.dll +0 -0
- mindspore/mindspore_shared_lib.dll +0 -0
- mindspore/msobj140.dll +0 -0
- mindspore/mspdb140.dll +0 -0
- mindspore/mspdbcore.dll +0 -0
- mindspore/mspdbst.dll +0 -0
- mindspore/mspft140.dll +0 -0
- mindspore/msvcdis140.dll +0 -0
- mindspore/msvcp140_1.dll +0 -0
- mindspore/msvcp140_2.dll +0 -0
- mindspore/msvcp140_atomic_wait.dll +0 -0
- mindspore/msvcp140_codecvt_ids.dll +0 -0
- mindspore/nn/__init__.py +1 -5
- mindspore/nn/cell.py +297 -234
- mindspore/nn/dynamic_lr.py +1 -1
- mindspore/nn/grad/cell_grad.py +17 -42
- mindspore/nn/layer/__init__.py +7 -4
- mindspore/nn/layer/activation.py +131 -88
- mindspore/nn/layer/basic.py +313 -613
- mindspore/nn/layer/channel_shuffle.py +103 -0
- mindspore/nn/layer/combined.py +1 -1
- mindspore/nn/layer/container.py +52 -6
- mindspore/nn/layer/conv.py +112 -43
- mindspore/nn/layer/dense.py +10 -9
- mindspore/nn/layer/embedding.py +36 -34
- mindspore/nn/layer/image.py +123 -27
- mindspore/nn/layer/math.py +108 -107
- mindspore/nn/layer/normalization.py +212 -366
- mindspore/nn/layer/padding.py +370 -42
- mindspore/nn/layer/pooling.py +1443 -219
- mindspore/nn/layer/rnn_cells.py +11 -16
- mindspore/nn/layer/rnns.py +38 -39
- mindspore/nn/layer/thor_layer.py +24 -25
- mindspore/nn/layer/timedistributed.py +5 -5
- mindspore/nn/layer/transformer.py +701 -0
- mindspore/nn/learning_rate_schedule.py +8 -8
- mindspore/nn/loss/__init__.py +9 -6
- mindspore/nn/loss/loss.py +678 -142
- mindspore/nn/metrics.py +53 -0
- mindspore/nn/optim/_dist_optimizer_registry.py +2 -2
- mindspore/nn/optim/ada_grad.py +8 -8
- mindspore/nn/optim/adadelta.py +2 -3
- mindspore/nn/optim/adafactor.py +18 -14
- mindspore/nn/optim/adam.py +429 -87
- mindspore/nn/optim/adamax.py +5 -6
- mindspore/nn/optim/adasum.py +10 -8
- mindspore/nn/optim/asgd.py +7 -7
- mindspore/nn/optim/ftrl.py +81 -11
- mindspore/nn/optim/lamb.py +7 -8
- mindspore/nn/optim/lars.py +4 -4
- mindspore/nn/optim/lazyadam.py +82 -7
- mindspore/nn/optim/momentum.py +8 -7
- mindspore/nn/optim/optimizer.py +19 -10
- mindspore/nn/optim/proximal_ada_grad.py +6 -5
- mindspore/nn/optim/rmsprop.py +3 -3
- mindspore/nn/optim/rprop.py +20 -16
- mindspore/nn/optim/sgd.py +21 -15
- mindspore/nn/optim/thor.py +23 -21
- mindspore/nn/probability/__init__.py +0 -2
- mindspore/nn/probability/bijector/bijector.py +7 -6
- mindspore/nn/probability/bijector/invert.py +4 -2
- mindspore/nn/probability/bijector/softplus.py +2 -2
- mindspore/nn/probability/bnn_layers/dense_variational.py +1 -1
- mindspore/nn/probability/bnn_layers/layer_distribution.py +2 -2
- mindspore/nn/probability/distribution/__init__.py +6 -0
- mindspore/nn/probability/distribution/_utils/custom_ops.py +3 -2
- mindspore/nn/probability/distribution/_utils/utils.py +11 -17
- mindspore/nn/probability/distribution/bernoulli.py +6 -6
- mindspore/nn/probability/distribution/beta.py +1 -1
- mindspore/nn/probability/distribution/categorical.py +9 -9
- mindspore/nn/probability/distribution/cauchy.py +8 -8
- mindspore/nn/probability/distribution/distribution.py +12 -6
- mindspore/nn/probability/distribution/exponential.py +5 -5
- mindspore/nn/probability/distribution/gamma.py +3 -3
- mindspore/nn/probability/distribution/geometric.py +6 -5
- mindspore/nn/probability/distribution/gumbel.py +5 -5
- mindspore/nn/probability/distribution/half_normal.py +133 -0
- mindspore/nn/probability/distribution/laplace.py +128 -0
- mindspore/nn/probability/distribution/log_normal.py +0 -1
- mindspore/nn/probability/distribution/logistic.py +4 -5
- mindspore/nn/probability/distribution/normal.py +11 -15
- mindspore/nn/probability/distribution/poisson.py +6 -2
- mindspore/nn/probability/distribution/student_t.py +150 -0
- mindspore/nn/probability/distribution/transformed_distribution.py +4 -4
- mindspore/nn/probability/distribution/uniform.py +5 -5
- mindspore/nn/reinforcement/_tensors_queue.py +3 -3
- mindspore/nn/reinforcement/tensor_array.py +2 -2
- mindspore/nn/sparse/sparse.py +8 -1
- mindspore/nn/wrap/cell_wrapper.py +55 -27
- mindspore/nn/wrap/grad_reducer.py +20 -11
- mindspore/nn/wrap/loss_scale.py +47 -30
- mindspore/numpy/array_creations.py +33 -22
- mindspore/numpy/array_ops.py +46 -42
- mindspore/numpy/logic_ops.py +6 -27
- mindspore/numpy/math_ops.py +26 -19
- mindspore/numpy/utils.py +1 -8
- mindspore/numpy/utils_const.py +112 -62
- mindspore/opencv_core452.dll +0 -0
- mindspore/opencv_imgcodecs452.dll +0 -0
- mindspore/opencv_imgproc452.dll +0 -0
- mindspore/ops/__init__.py +6 -3
- mindspore/ops/_constants.py +0 -6
- mindspore/ops/_grad/__init__.py +2 -1
- mindspore/ops/_grad/grad_array_ops.py +209 -152
- mindspore/ops/_grad/grad_base.py +55 -17
- mindspore/ops/_grad/grad_clip_ops.py +11 -3
- mindspore/ops/_grad/grad_comm_ops.py +58 -47
- mindspore/ops/_grad/grad_implementations.py +21 -61
- mindspore/ops/_grad/grad_inner_ops.py +48 -6
- mindspore/ops/_grad/grad_math_ops.py +306 -161
- mindspore/ops/_grad/grad_nn_ops.py +192 -181
- mindspore/ops/_grad/grad_other_ops.py +1 -1
- mindspore/ops/_grad/grad_quant_ops.py +5 -5
- mindspore/ops/_grad/grad_sequence_ops.py +296 -0
- mindspore/ops/_grad/grad_sparse.py +15 -9
- mindspore/ops/_grad_experimental/__init__.py +1 -0
- mindspore/ops/_grad_experimental/grad_array_ops.py +441 -55
- mindspore/ops/_grad_experimental/grad_image_ops.py +25 -7
- mindspore/ops/_grad_experimental/grad_inner_ops.py +3 -44
- mindspore/ops/_grad_experimental/grad_linalg_ops.py +16 -21
- mindspore/ops/_grad_experimental/grad_math_ops.py +979 -49
- mindspore/ops/_grad_experimental/grad_nn_ops.py +78 -8
- mindspore/ops/_grad_experimental/grad_scalar_ops.py +112 -0
- mindspore/ops/_grad_experimental/grad_sparse_ops.py +197 -13
- mindspore/ops/_op_impl/__init__.py +3 -3
- mindspore/ops/_op_impl/_custom_op/__init__.py +0 -1
- mindspore/ops/_op_impl/_custom_op/_basic.py +0 -1
- mindspore/ops/_op_impl/_custom_op/batch_matmul_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold.py +4 -2
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold2.py +2 -2
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py +5 -5
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold_grad.py +3 -3
- mindspore/ops/_op_impl/_custom_op/cholesky_trsm_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/correction_mul.py +3 -3
- mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/dsd_back_impl.py +4 -8
- mindspore/ops/_op_impl/_custom_op/dsd_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad_reduce.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad_reduce.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fused_abs_max1_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/img2col_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/matmul_cube_dense_left_impl.py +2 -2
- mindspore/ops/_op_impl/_custom_op/matmul_cube_dense_right_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_left_cast_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_right_mul_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/matmul_cube_impl.py +2 -2
- mindspore/ops/_op_impl/_custom_op/matmul_dds_grad_impl.py +0 -1
- mindspore/ops/_op_impl/_custom_op/matmul_dds_impl.py +0 -1
- mindspore/ops/_op_impl/_custom_op/matrix_combine_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/minmax_update_perchannel.py +2 -2
- mindspore/ops/_op_impl/_custom_op/minmax_update_perlayer.py +2 -2
- mindspore/ops/_op_impl/_custom_op/transpose02314_impl.py +1 -1
- mindspore/ops/_op_impl/aicpu/__init__.py +238 -3
- mindspore/ops/_op_impl/aicpu/abs.py +36 -0
- mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_2d.py +34 -0
- mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_2d_grad.py +34 -0
- mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_3d.py +39 -0
- mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_3d_grad.py +39 -0
- mindspore/ops/_op_impl/aicpu/adaptive_max_pool_2d_grad.py +37 -0
- mindspore/ops/_op_impl/aicpu/adaptive_max_pool_3d.py +42 -0
- mindspore/ops/_op_impl/aicpu/adaptive_max_pool_3d_grad.py +152 -0
- mindspore/ops/_op_impl/aicpu/add.py +43 -0
- mindspore/ops/_op_impl/aicpu/addcdiv.py +0 -32
- mindspore/ops/_op_impl/aicpu/addcmul.py +0 -84
- mindspore/ops/_op_impl/aicpu/affine_grid_grad.py +35 -0
- mindspore/ops/_op_impl/aicpu/arg_max.py +75 -0
- mindspore/ops/_op_impl/aicpu/arg_min.py +75 -0
- mindspore/ops/_op_impl/aicpu/argmin_with_value.py +43 -0
- mindspore/ops/_op_impl/aicpu/batch_matmul.py +43 -0
- mindspore/ops/_op_impl/aicpu/batch_norm_grad_grad.py +49 -0
- mindspore/ops/_op_impl/aicpu/bernoulli.py +48 -0
- mindspore/ops/_op_impl/aicpu/bessel_i0.py +31 -0
- mindspore/ops/_op_impl/aicpu/bias_add.py +44 -0
- mindspore/ops/_op_impl/aicpu/bias_add_grad.py +43 -0
- mindspore/ops/_op_impl/aicpu/bincount.py +33 -0
- mindspore/{nn/probability/infer/variational/__init__.py → ops/_op_impl/aicpu/cauchy.py} +17 -10
- mindspore/ops/_op_impl/aicpu/channel_shuffle.py +40 -0
- mindspore/ops/_op_impl/aicpu/cholesky.py +1 -1
- mindspore/ops/_op_impl/{cpu/bias_add.py → aicpu/choleskygrad.py} +9 -7
- mindspore/ops/_op_impl/aicpu/combined_non_max_suppression.py +42 -0
- mindspore/ops/_op_impl/aicpu/concat_offset.py +42 -0
- mindspore/ops/_op_impl/aicpu/concat_offset_v1.py +31 -0
- mindspore/ops/_op_impl/aicpu/conj.py +11 -0
- mindspore/ops/_op_impl/aicpu/crop_and_resize_grad_image.py +38 -0
- mindspore/ops/_op_impl/aicpu/cumulative_logsumexp.py +36 -0
- mindspore/ops/_op_impl/aicpu/deformable_offsets.py +38 -0
- mindspore/ops/_op_impl/aicpu/deformable_offsets_grad.py +2 -2
- mindspore/ops/_op_impl/aicpu/dense_to_sparse_set_operation.py +48 -0
- mindspore/ops/_op_impl/aicpu/diag.py +36 -0
- mindspore/ops/_op_impl/aicpu/diag_part.py +36 -0
- mindspore/ops/_op_impl/aicpu/diagonal.py +35 -0
- mindspore/ops/_op_impl/{cpu/bias_add_grad.py → aicpu/digamma.py} +9 -7
- mindspore/ops/_op_impl/aicpu/eig.py +35 -0
- mindspore/ops/_op_impl/aicpu/fft_with_size.py +41 -0
- mindspore/ops/_op_impl/aicpu/flatten.py +1 -0
- mindspore/ops/_op_impl/aicpu/fmax.py +36 -0
- mindspore/ops/_op_impl/aicpu/fmin.py +37 -0
- mindspore/ops/_op_impl/aicpu/fractional_max_pool3d_with_fixed_ksize.py +1 -1
- mindspore/ops/_op_impl/aicpu/fse_decode.py +43 -0
- mindspore/ops/_op_impl/aicpu/glu.py +33 -0
- mindspore/ops/_op_impl/aicpu/glu_grad.py +34 -0
- mindspore/ops/_op_impl/aicpu/greater.py +41 -0
- mindspore/ops/_op_impl/aicpu/greater_equal.py +41 -0
- mindspore/ops/_op_impl/aicpu/index_put.py +50 -0
- mindspore/ops/_op_impl/{tbe/scatter_add_ds.py → aicpu/inplace_index_add.py} +17 -21
- mindspore/ops/_op_impl/aicpu/instance_norm_v2.py +41 -0
- mindspore/ops/_op_impl/aicpu/instance_norm_v2_grad.py +44 -0
- mindspore/ops/_op_impl/aicpu/layer_norm_grad_grad.py +47 -0
- mindspore/ops/_op_impl/aicpu/less.py +41 -0
- mindspore/ops/_op_impl/aicpu/less_equal.py +41 -0
- mindspore/ops/_op_impl/aicpu/lgamma.py +32 -0
- mindspore/ops/_op_impl/aicpu/log_normal_reverse.py +33 -0
- mindspore/ops/_op_impl/aicpu/logit.py +33 -0
- mindspore/ops/_op_impl/aicpu/logit_grad.py +34 -0
- mindspore/ops/_op_impl/aicpu/masked_fill.py +42 -0
- mindspore/ops/_op_impl/aicpu/masked_scatter.py +39 -0
- mindspore/ops/_op_impl/aicpu/matmul.py +39 -0
- mindspore/ops/_op_impl/aicpu/matrix_logarithm.py +31 -0
- mindspore/ops/_op_impl/aicpu/matrix_power.py +32 -0
- mindspore/ops/_op_impl/aicpu/matrix_solve_ls.py +36 -0
- mindspore/ops/_op_impl/aicpu/matrix_triangular_solve.py +36 -0
- mindspore/ops/_op_impl/aicpu/mirror_pad.py +2 -0
- mindspore/ops/_op_impl/aicpu/mirror_pad_grad.py +0 -4
- mindspore/ops/_op_impl/aicpu/mul.py +3 -1
- mindspore/ops/_op_impl/aicpu/multinomial.py +14 -6
- mindspore/ops/_op_impl/aicpu/multinomial_with_replacement.py +35 -0
- mindspore/ops/_op_impl/aicpu/nan_to_num.py +34 -0
- mindspore/ops/_op_impl/aicpu/nllloss.py +38 -0
- mindspore/ops/_op_impl/aicpu/nllloss_grad.py +39 -0
- mindspore/ops/_op_impl/aicpu/ones_like.py +0 -2
- mindspore/ops/_op_impl/aicpu/polar.py +32 -0
- mindspore/ops/_op_impl/aicpu/polygamma.py +34 -0
- mindspore/ops/_op_impl/aicpu/qr.py +36 -0
- mindspore/ops/_op_impl/aicpu/quant_dtype_cast.py +40 -0
- mindspore/ops/_op_impl/aicpu/quantile.py +35 -0
- mindspore/ops/_op_impl/aicpu/ragged_tensor_to_sparse.py +73 -0
- mindspore/ops/_op_impl/aicpu/ragged_tensor_to_tensor.py +74 -0
- mindspore/ops/_op_impl/aicpu/random_shuffle.py +3 -0
- mindspore/ops/_op_impl/aicpu/randperm_v2.py +41 -0
- mindspore/ops/_op_impl/aicpu/range.py +36 -0
- mindspore/ops/_op_impl/aicpu/reciprocal.py +34 -0
- mindspore/ops/_op_impl/aicpu/reciprocal_grad.py +35 -0
- mindspore/ops/_op_impl/aicpu/reduce_sum.py +57 -0
- mindspore/ops/_op_impl/aicpu/resize_bicubic.py +2 -8
- mindspore/ops/_op_impl/aicpu/resize_bicubic_grad.py +1 -1
- mindspore/ops/_op_impl/aicpu/resize_v2.py +68 -0
- mindspore/ops/_op_impl/aicpu/resize_v2_grad.py +68 -0
- mindspore/ops/_op_impl/aicpu/scatter_elements.py +4 -0
- mindspore/ops/_op_impl/aicpu/scatter_nd_update.py +2 -0
- mindspore/ops/_op_impl/aicpu/search_sorted.py +12 -6
- mindspore/ops/_op_impl/aicpu/self_adjoint_eig.py +34 -0
- mindspore/ops/_op_impl/aicpu/sequence_add.py +34 -0
- mindspore/ops/_op_impl/aicpu/sequence_add_offset.py +34 -0
- mindspore/ops/_op_impl/aicpu/sequence_addn.py +38 -0
- mindspore/ops/_op_impl/aicpu/slice_grad.py +76 -0
- mindspore/ops/_op_impl/aicpu/smooth_l1_loss.py +35 -0
- mindspore/ops/_op_impl/aicpu/smooth_l1_loss_grad.py +37 -0
- mindspore/ops/_op_impl/aicpu/sort.py +39 -0
- mindspore/ops/_op_impl/aicpu/sparse_apply_adagrad_da.py +0 -24
- mindspore/ops/_op_impl/aicpu/sparse_cross.py +42 -0
- mindspore/ops/_op_impl/aicpu/sparse_fill_empty_rows.py +63 -0
- mindspore/ops/_op_impl/aicpu/sparse_fill_empty_rows_grad.py +45 -0
- mindspore/ops/_op_impl/aicpu/sparse_matrix_mat_mul.py +56 -0
- mindspore/ops/_op_impl/{tbe/slice_ds.py → aicpu/sparse_segment_sum.py} +16 -24
- mindspore/ops/_op_impl/aicpu/sparse_segment_sum_with_num_segments.py +68 -0
- mindspore/ops/_op_impl/aicpu/sparse_slice.py +63 -0
- mindspore/ops/_op_impl/aicpu/sparse_slice_grad.py +61 -0
- mindspore/ops/_op_impl/aicpu/squared_difference.py +2 -0
- mindspore/ops/_op_impl/aicpu/strided_slice_v2.py +93 -0
- mindspore/ops/_op_impl/aicpu/strided_slice_v2_grad.py +66 -0
- mindspore/ops/_op_impl/aicpu/tensor_scatter_update.py +59 -0
- mindspore/ops/_op_impl/{tbe/gather_v2.py → aicpu/tile.py} +24 -24
- mindspore/ops/_op_impl/aicpu/tridiagonal_solve.py +35 -0
- mindspore/ops/_op_impl/aicpu/tril_indices.py +34 -0
- mindspore/ops/_op_impl/aicpu/triu_indices.py +34 -0
- mindspore/ops/_op_impl/aicpu/uniform.py +34 -0
- mindspore/ops/_op_impl/aicpu/uniform_candidate_sampler.py +1 -0
- mindspore/ops/_op_impl/aicpu/unique_consecutive.py +10 -2
- mindspore/ops/_op_impl/cpu/__init__.py +1 -2
- mindspore/ops/_op_impl/cpu/dynamic_shape.py +5 -1
- mindspore/ops/_op_impl/cpu/maximum_grad.py +2 -0
- mindspore/{compression/common/__init__.py → ops/_op_impl/cpu/pyexecute.py} +13 -8
- mindspore/ops/_op_impl/cpu/reduce_sum.py +8 -0
- mindspore/ops/_op_impl/cpu/sparse_slice.py +62 -0
- mindspore/ops/_op_impl/cpu/sparse_slice_grad.py +60 -0
- mindspore/ops/_op_impl/cpu/tensor_shape.py +5 -1
- mindspore/ops/_op_impl/tbe/__init__.py +27 -608
- mindspore/ops/_op_impl/tbe/addcdiv_ds.py +42 -0
- mindspore/ops/_op_impl/tbe/addcmul_ds.py +44 -0
- mindspore/ops/_op_impl/tbe/assign_add_ds.py +1 -0
- mindspore/ops/_op_impl/tbe/atomic_addr_clean.py +1 -1
- mindspore/ops/_op_impl/tbe/avg_pool_3d_grad.py +1 -1
- mindspore/ops/_op_impl/tbe/basic_lstm_cell_c_state_grad_v2.py +0 -1
- mindspore/ops/_op_impl/tbe/batch_to_space.py +1 -1
- mindspore/ops/_op_impl/tbe/batch_to_space_nd.py +1 -1
- mindspore/ops/_op_impl/tbe/batch_to_space_nd_v2.py +41 -0
- mindspore/ops/_op_impl/tbe/bce_with_logits_loss.py +1 -0
- mindspore/ops/_op_impl/tbe/bias_add_grad.py +2 -0
- mindspore/ops/_op_impl/tbe/bn_infer_grad.py +4 -2
- mindspore/ops/_op_impl/tbe/bn_infer_grad_ds.py +40 -0
- mindspore/ops/_op_impl/tbe/bn_training_update.py +0 -1
- mindspore/ops/_op_impl/tbe/bn_training_update_ds.py +0 -1
- mindspore/ops/_op_impl/tbe/broadcast_to_ds.py +6 -4
- mindspore/ops/_op_impl/tbe/cast.py +0 -2
- mindspore/ops/_op_impl/tbe/cast_ds.py +3 -3
- mindspore/ops/_op_impl/tbe/ctc_loss_v2.py +0 -2
- mindspore/ops/_op_impl/tbe/ctc_loss_v2_grad.py +0 -2
- mindspore/ops/_op_impl/tbe/data_format_dim_map_ds.py +1 -0
- mindspore/ops/_op_impl/tbe/deformable_offsets.py +1 -0
- mindspore/ops/_op_impl/tbe/depthwise_conv2d.py +1 -1
- mindspore/ops/_op_impl/tbe/dynamic_atomic_addr_clean.py +1 -1
- mindspore/ops/_op_impl/tbe/gather_nd.py +1 -0
- mindspore/ops/_op_impl/tbe/greater.py +2 -0
- mindspore/ops/_op_impl/tbe/{index_add.py → inplace_index_add.py} +3 -6
- mindspore/ops/_op_impl/tbe/layer_norm_beta_gamma_backprop_v2.py +0 -1
- mindspore/ops/_op_impl/tbe/npu_clear_float_status_v2.py +35 -0
- mindspore/ops/_op_impl/tbe/npu_get_float_status_v2.py +35 -0
- mindspore/ops/_op_impl/tbe/one_hot_ds.py +0 -6
- mindspore/ops/_op_impl/tbe/{greater_ds.py → reduce_all_ds.py} +13 -16
- mindspore/ops/_op_impl/tbe/reduce_any_ds.py +39 -0
- mindspore/ops/_op_impl/tbe/roi_align_ds.py +44 -0
- mindspore/ops/_op_impl/tbe/roi_align_grad_ds.py +44 -0
- mindspore/ops/_op_impl/tbe/scatter_add.py +2 -0
- mindspore/ops/_op_impl/tbe/scatter_nd_add.py +2 -2
- mindspore/ops/_op_impl/tbe/slice.py +26 -15
- mindspore/ops/_op_impl/tbe/space_to_batch.py +1 -1
- mindspore/ops/_op_impl/tbe/space_to_batch_nd.py +1 -1
- mindspore/ops/_op_impl/tbe/strided_slice_grad_d.py +1 -0
- mindspore/ops/_op_impl/tbe/trans_data_ds.py +15 -5
- mindspore/ops/_op_impl/tbe/unsorted_segment_sum.py +1 -1
- mindspore/ops/_op_impl/tbe/unsorted_segment_sum_ds.py +2 -0
- mindspore/ops/_primitive_cache.py +3 -2
- mindspore/ops/_register_for_op.py +11 -0
- mindspore/ops/_utils/__init__.py +1 -1
- mindspore/ops/_utils/utils.py +20 -41
- mindspore/ops/_vmap/__init__.py +2 -2
- mindspore/ops/_vmap/vmap_array_ops.py +170 -78
- mindspore/ops/_vmap/vmap_base.py +24 -10
- mindspore/ops/_vmap/vmap_convolution_ops.py +7 -10
- mindspore/ops/_vmap/vmap_grad_math_ops.py +4 -4
- mindspore/ops/_vmap/vmap_grad_nn_ops.py +41 -9
- mindspore/ops/_vmap/vmap_image_ops.py +52 -0
- mindspore/ops/_vmap/vmap_math_ops.py +77 -6
- mindspore/ops/_vmap/vmap_nn_ops.py +78 -29
- mindspore/ops/_vmap/vmap_other_ops.py +3 -1
- mindspore/ops/_vmap/vmap_random_ops.py +55 -3
- mindspore/ops/_vmap/vmap_sparse_ops.py +1 -0
- mindspore/ops/bprop_mindir/AdaptiveAvgPool2D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/AdaptiveMaxPool2D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ApproximateEqual_bprop.mindir +18 -19
- mindspore/ops/bprop_mindir/Argmax_bprop.mindir +13 -12
- mindspore/ops/bprop_mindir/Argmin_bprop.mindir +14 -13
- mindspore/ops/bprop_mindir/AssignSub_bprop.mindir +17 -18
- mindspore/ops/bprop_mindir/Assign_bprop.mindir +16 -16
- mindspore/ops/bprop_mindir/AvgPool3D_bprop.mindir +150 -0
- mindspore/ops/bprop_mindir/AvgPool_bprop.mindir +66 -0
- mindspore/ops/bprop_mindir/BCEWithLogitsLoss_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/BNTrainingReduce_bprop.mindir +13 -12
- mindspore/ops/bprop_mindir/BatchNormGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/BatchToSpaceND_bprop.mindir +28 -0
- mindspore/ops/bprop_mindir/BiasAddGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/BinaryCrossEntropy_bprop.mindir +33 -0
- mindspore/ops/bprop_mindir/BroadcastTo_bprop.mindir +306 -0
- mindspore/ops/bprop_mindir/Broadcast_bprop.mindir +12 -8
- mindspore/ops/bprop_mindir/CTCLoss_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Concat_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Conv2DBackpropFilter_bprop.mindir +240 -0
- mindspore/ops/bprop_mindir/Conv2DBackpropInput_bprop.mindir +247 -0
- mindspore/ops/bprop_mindir/Conv2DTranspose_bprop.mindir +247 -0
- mindspore/ops/bprop_mindir/Conv3DTranspose_bprop.mindir +315 -0
- mindspore/ops/bprop_mindir/Conv3D_bprop.mindir +278 -0
- mindspore/ops/bprop_mindir/DType_bprop.mindir +12 -12
- mindspore/ops/bprop_mindir/DeformableOffsets_bprop.mindir +58 -0
- mindspore/ops/bprop_mindir/Depend_bprop.mindir +12 -13
- mindspore/ops/bprop_mindir/DepthToSpace_bprop.mindir +23 -0
- mindspore/ops/bprop_mindir/DepthwiseConv2dNative_bprop.mindir +138 -0
- mindspore/ops/bprop_mindir/DiagPart_bprop.mindir +15 -0
- mindspore/ops/bprop_mindir/Dropout2D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Dropout3D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DropoutDoMask_bprop.mindir +22 -24
- mindspore/ops/bprop_mindir/DropoutGenMask_bprop.mindir +16 -14
- mindspore/ops/bprop_mindir/DropoutGrad_bprop.mindir +27 -0
- mindspore/ops/bprop_mindir/Dropout_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DynamicGRUV2_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DynamicRNN_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DynamicShape_bprop.mindir +12 -12
- mindspore/ops/bprop_mindir/Elu_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/EmbeddingLookup_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Equal_bprop.mindir +18 -19
- mindspore/ops/bprop_mindir/ExpandDims_bprop.mindir +58 -0
- mindspore/ops/bprop_mindir/FastGeLU_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/Flatten_bprop.mindir +54 -0
- mindspore/ops/bprop_mindir/FloorDiv_bprop.mindir +18 -15
- mindspore/ops/bprop_mindir/GatherD_bprop.mindir +26 -0
- mindspore/ops/bprop_mindir/GatherNd_bprop.mindir +57 -0
- mindspore/ops/bprop_mindir/Gather_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/GreaterEqual_bprop.mindir +17 -18
- mindspore/ops/bprop_mindir/Greater_bprop.mindir +18 -19
- mindspore/ops/bprop_mindir/HSigmoid_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/HSwish_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/IOU_bprop.mindir +18 -19
- mindspore/ops/bprop_mindir/InstanceNorm_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/IsFinite_bprop.mindir +13 -12
- mindspore/ops/bprop_mindir/IsInf_bprop.mindir +13 -10
- mindspore/ops/bprop_mindir/IsNan_bprop.mindir +14 -11
- mindspore/ops/bprop_mindir/KLDivLoss_bprop.mindir +126 -0
- mindspore/ops/bprop_mindir/L2Loss_bprop.mindir +15 -0
- mindspore/ops/bprop_mindir/L2Normalize_bprop.mindir +30 -0
- mindspore/ops/bprop_mindir/LRN_bprop.mindir +43 -0
- mindspore/ops/bprop_mindir/LayerNormGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/LessEqual_bprop.mindir +18 -19
- mindspore/ops/bprop_mindir/Less_bprop.mindir +17 -18
- mindspore/ops/bprop_mindir/LinSpace_bprop.mindir +22 -19
- mindspore/ops/bprop_mindir/Load_bprop.mindir +12 -13
- mindspore/ops/bprop_mindir/LogSoftmax_bprop.mindir +23 -0
- mindspore/ops/bprop_mindir/LogicalAnd_bprop.mindir +17 -18
- mindspore/ops/bprop_mindir/LogicalNot_bprop.mindir +14 -13
- mindspore/ops/bprop_mindir/MaskedSelect_bprop.mindir +21 -0
- mindspore/ops/bprop_mindir/MaxPool3DGradGrad_bprop.mindir +74 -0
- mindspore/ops/bprop_mindir/MaxPool3DGrad_bprop.mindir +74 -0
- mindspore/ops/bprop_mindir/MaxPool3D_bprop.mindir +75 -0
- mindspore/ops/bprop_mindir/MaxPoolGradGrad_bprop.mindir +65 -0
- mindspore/ops/bprop_mindir/MaxPoolWithArgmax_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Maximum_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Minimum_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/MirrorPad_bprop.mindir +27 -0
- mindspore/ops/bprop_mindir/Mish_bprop.mindir +35 -0
- mindspore/ops/bprop_mindir/MulNoNan_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/NLLLoss_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/NonZero_bprop.mindir +14 -0
- mindspore/ops/bprop_mindir/NotEqual_bprop.mindir +18 -19
- mindspore/ops/bprop_mindir/OneHot_bprop.mindir +25 -23
- mindspore/ops/bprop_mindir/OnesLike_bprop.mindir +13 -13
- mindspore/ops/bprop_mindir/PReLU_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Pad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Padding_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/RNNTLoss_bprop.mindir +29 -0
- mindspore/ops/bprop_mindir/ROIAlign_bprop.mindir +82 -0
- mindspore/ops/bprop_mindir/Range_bprop.mindir +21 -19
- mindspore/ops/bprop_mindir/Rank_bprop.mindir +11 -11
- mindspore/ops/bprop_mindir/ReLU6_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/ReLUV2_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ReduceAll_bprop.mindir +18 -17
- mindspore/ops/bprop_mindir/ReduceAny_bprop.mindir +18 -17
- mindspore/ops/bprop_mindir/ReluGrad_bprop.mindir +19 -23
- mindspore/ops/bprop_mindir/Reshape_bprop.mindir +60 -0
- mindspore/ops/bprop_mindir/ResizeBilinear_bprop.mindir +29 -0
- mindspore/ops/bprop_mindir/ResizeNearestNeighbor_bprop.mindir +89 -0
- mindspore/ops/bprop_mindir/ReverseSequence_bprop.mindir +52 -0
- mindspore/ops/bprop_mindir/ReverseV2_bprop.mindir +22 -0
- mindspore/ops/bprop_mindir/Round_bprop.mindir +14 -13
- mindspore/ops/bprop_mindir/ScatterMax_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ScatterMin_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ScatterNdUpdate_bprop.mindir +22 -0
- mindspore/ops/bprop_mindir/ScatterNd_bprop.mindir +24 -0
- mindspore/ops/bprop_mindir/ScatterNonAliasingAdd_bprop.mindir +22 -0
- mindspore/ops/bprop_mindir/ScatterUpdate_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/SeLU_bprop.mindir +21 -0
- mindspore/ops/bprop_mindir/Select_bprop.mindir +30 -34
- mindspore/ops/bprop_mindir/Shape_bprop.mindir +12 -12
- mindspore/ops/bprop_mindir/SigmoidCrossEntropyWithLogits_bprop.mindir +21 -0
- mindspore/ops/bprop_mindir/SigmoidGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Sigmoid_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/Sign_bprop.mindir +13 -12
- mindspore/ops/bprop_mindir/Slice_bprop.mindir +26 -0
- mindspore/ops/bprop_mindir/SmoothL1Loss_bprop.mindir +36 -0
- mindspore/ops/bprop_mindir/SoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Softplus_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/Softsign_bprop.mindir +33 -0
- mindspore/ops/bprop_mindir/Sort_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/SpaceToBatchND_bprop.mindir +28 -0
- mindspore/ops/bprop_mindir/SpaceToDepth_bprop.mindir +23 -0
- mindspore/ops/bprop_mindir/SparseGatherV2_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/SparseSoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Split_bprop.mindir +22 -0
- mindspore/ops/bprop_mindir/Squeeze_bprop.mindir +54 -0
- mindspore/ops/bprop_mindir/StridedSliceGrad_bprop.mindir +95 -0
- mindspore/ops/bprop_mindir/StridedSlice_bprop.mindir +98 -0
- mindspore/ops/bprop_mindir/Switch_bprop.mindir +28 -32
- mindspore/ops/bprop_mindir/TanhGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Tanh_bprop.mindir +66 -0
- mindspore/ops/bprop_mindir/TensorScatterAdd_bprop.mindir +22 -0
- mindspore/ops/bprop_mindir/TensorScatterUpdate_bprop.mindir +29 -0
- mindspore/ops/bprop_mindir/TensorShape_bprop.mindir +14 -0
- mindspore/ops/bprop_mindir/Tile_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/TopK_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/TransShape_bprop.mindir +23 -0
- mindspore/ops/bprop_mindir/TruncateDiv_bprop.mindir +18 -15
- mindspore/ops/bprop_mindir/TupleGetItem_bprop.mindir +11 -13
- mindspore/ops/bprop_mindir/Unique_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/Unstack_bprop.mindir +22 -0
- mindspore/ops/bprop_mindir/UpsampleNearest3D_bprop.mindir +32 -0
- mindspore/ops/bprop_mindir/UpsampleTrilinear3D_bprop.mindir +38 -0
- mindspore/ops/bprop_mindir/ZerosLike_bprop.mindir +13 -12
- mindspore/ops/bprop_mindir/__init__.py +1 -4
- mindspore/ops/bprop_mindir/generate_mindir.py +32 -20
- mindspore/ops/composite/__init__.py +12 -13
- mindspore/ops/composite/base.py +261 -254
- mindspore/ops/composite/env_ops.py +41 -0
- mindspore/ops/composite/math_ops.py +197 -156
- mindspore/ops/composite/multitype_ops/_compile_utils.py +428 -176
- mindspore/ops/composite/multitype_ops/_constexpr_utils.py +188 -87
- mindspore/ops/composite/multitype_ops/add_impl.py +23 -1
- mindspore/ops/composite/multitype_ops/div_impl.py +3 -3
- mindspore/ops/composite/multitype_ops/equal_impl.py +1 -0
- mindspore/ops/composite/multitype_ops/floordiv_impl.py +1 -1
- mindspore/ops/composite/multitype_ops/getitem_impl.py +52 -5
- mindspore/ops/composite/multitype_ops/greater_equal_impl.py +31 -0
- mindspore/ops/composite/multitype_ops/greater_impl.py +31 -0
- mindspore/ops/composite/multitype_ops/in_impl.py +15 -3
- mindspore/ops/composite/multitype_ops/less_equal_impl.py +33 -2
- mindspore/ops/composite/multitype_ops/less_impl.py +33 -0
- mindspore/ops/composite/multitype_ops/logical_and_impl.py +2 -2
- mindspore/ops/composite/multitype_ops/logical_or_impl.py +2 -1
- mindspore/ops/composite/multitype_ops/mod_impl.py +1 -1
- mindspore/ops/composite/multitype_ops/mul_impl.py +21 -7
- mindspore/ops/composite/multitype_ops/not_in_impl.py +15 -3
- mindspore/ops/composite/multitype_ops/ones_like_impl.py +2 -4
- mindspore/ops/composite/multitype_ops/pow_impl.py +1 -0
- mindspore/ops/composite/multitype_ops/setitem_impl.py +62 -70
- mindspore/ops/composite/multitype_ops/sub_impl.py +3 -3
- mindspore/ops/composite/multitype_ops/zeros_like_impl.py +41 -4
- mindspore/ops/function/__init__.py +323 -8
- mindspore/ops/function/array_func.py +3511 -780
- mindspore/ops/function/clip_func.py +329 -0
- mindspore/ops/function/debug_func.py +6 -6
- mindspore/ops/function/grad/__init__.py +5 -1
- mindspore/ops/function/grad/grad_func.py +736 -65
- mindspore/ops/function/image_func.py +270 -0
- mindspore/ops/function/linalg_func.py +268 -8
- mindspore/ops/function/math_func.py +8032 -3164
- mindspore/ops/function/nn_func.py +5619 -1855
- mindspore/ops/function/other_func.py +115 -0
- mindspore/ops/function/parameter_func.py +11 -10
- mindspore/ops/function/random_func.py +939 -77
- mindspore/ops/function/sparse_func.py +249 -84
- mindspore/ops/function/sparse_unary_func.py +2303 -0
- mindspore/ops/function/spectral_func.py +146 -0
- mindspore/ops/function/vmap_func.py +114 -0
- mindspore/ops/functional.py +182 -254
- mindspore/ops/op_info_register.py +79 -34
- mindspore/ops/operations/__init__.py +210 -118
- mindspore/ops/operations/_csr_ops.py +7 -7
- mindspore/ops/operations/_embedding_cache_ops.py +25 -15
- mindspore/ops/operations/_grad_ops.py +447 -322
- mindspore/ops/operations/_inner_ops.py +547 -176
- mindspore/ops/operations/_map_tensor_ops.py +112 -0
- mindspore/ops/operations/_ms_kernel.py +29 -27
- mindspore/ops/operations/_ocr_ops.py +11 -11
- mindspore/ops/operations/_opaque_predicate_registry.py +41 -0
- mindspore/ops/operations/_quant_ops.py +186 -101
- mindspore/ops/operations/_rl_inner_ops.py +122 -61
- mindspore/ops/operations/_scalar_ops.py +466 -0
- mindspore/ops/operations/_sequence_ops.py +1047 -0
- mindspore/ops/operations/_tensor_array.py +10 -11
- mindspore/ops/operations/_thor_ops.py +4 -4
- mindspore/ops/operations/array_ops.py +1428 -1226
- mindspore/ops/operations/comm_ops.py +180 -117
- mindspore/ops/operations/control_ops.py +4 -2
- mindspore/ops/operations/custom_ops.py +185 -98
- mindspore/ops/operations/debug_ops.py +92 -54
- mindspore/ops/operations/image_ops.py +406 -211
- mindspore/ops/operations/inner_ops.py +42 -53
- mindspore/ops/operations/linalg_ops.py +32 -29
- mindspore/ops/operations/math_ops.py +2076 -897
- mindspore/ops/operations/nn_ops.py +1282 -1252
- mindspore/ops/operations/other_ops.py +124 -278
- mindspore/ops/operations/random_ops.py +345 -178
- mindspore/ops/operations/rl_ops.py +8 -9
- mindspore/ops/operations/sparse_ops.py +502 -157
- mindspore/ops/operations/spectral_ops.py +107 -0
- mindspore/ops/primitive.py +192 -15
- mindspore/ops/vm_impl_registry.py +23 -2
- mindspore/parallel/__init__.py +6 -1
- mindspore/parallel/_auto_parallel_context.py +199 -92
- mindspore/parallel/_cell_wrapper.py +4 -2
- mindspore/parallel/_cost_model_context.py +3 -0
- mindspore/parallel/_dp_allreduce_fusion.py +2 -1
- mindspore/parallel/_offload_context.py +185 -0
- mindspore/parallel/_parallel_serialization.py +167 -28
- mindspore/parallel/_ps_context.py +9 -5
- mindspore/parallel/_recovery_context.py +1 -1
- mindspore/parallel/_tensor.py +9 -1
- mindspore/{nn/transformer → parallel/_transformer}/__init__.py +6 -6
- mindspore/{nn/transformer → parallel/_transformer}/layers.py +59 -37
- mindspore/{nn/transformer → parallel/_transformer}/loss.py +4 -7
- mindspore/{nn/transformer → parallel/_transformer}/moe.py +160 -35
- mindspore/{nn/transformer → parallel/_transformer}/op_parallel_config.py +3 -3
- mindspore/{nn/transformer → parallel/_transformer}/transformer.py +235 -196
- mindspore/parallel/_utils.py +47 -7
- mindspore/parallel/algo_parameter_config.py +5 -1
- mindspore/parallel/checkpoint_transform.py +329 -0
- mindspore/parallel/shard.py +229 -0
- mindspore/perf_msvcbuildinsights.dll +0 -0
- mindspore/pgodb140.dll +0 -0
- mindspore/pgort140.dll +0 -0
- mindspore/profiler/__init__.py +2 -1
- mindspore/profiler/common/util.py +4 -3
- mindspore/profiler/common/validator/validate_path.py +2 -2
- mindspore/profiler/envprofiling.py +249 -0
- mindspore/profiler/parser/aicpu_data_parser.py +38 -39
- mindspore/profiler/parser/ascend_timeline_generator.py +497 -0
- mindspore/profiler/parser/base_timeline_generator.py +471 -0
- mindspore/profiler/parser/cpu_gpu_timeline_generator.py +684 -0
- mindspore/profiler/parser/framework_parser.py +42 -16
- mindspore/profiler/parser/hccl_parser.py +158 -158
- mindspore/profiler/parser/hwts_log_parser.py +7 -6
- mindspore/profiler/parser/integrator.py +18 -1579
- mindspore/profiler/parser/minddata_analyzer.py +8 -8
- mindspore/profiler/parser/msadvisor_analyzer.py +14 -27
- mindspore/profiler/parser/msadvisor_parser.py +2 -4
- mindspore/profiler/parser/optime_parser.py +17 -18
- mindspore/profiler/parser/profiler_info.py +108 -0
- mindspore/profiler/parser/step_trace_parser.py +1 -1
- mindspore/profiler/profiling.py +396 -194
- mindspore/rewrite/__init__.py +6 -2
- mindspore/rewrite/api/node.py +51 -110
- mindspore/rewrite/api/node_type.py +10 -6
- mindspore/rewrite/api/pattern_engine.py +51 -7
- mindspore/rewrite/api/scoped_value.py +64 -53
- mindspore/rewrite/api/symbol_tree.py +108 -61
- mindspore/rewrite/api/tree_node_helper.py +2 -3
- mindspore/{compression/quant/__init__.py → rewrite/ast_creator_register.py} +20 -11
- mindspore/rewrite/ast_helpers/__init__.py +6 -3
- mindspore/rewrite/ast_helpers/ast_creator.py +115 -0
- mindspore/rewrite/ast_helpers/ast_finder.py +99 -1
- mindspore/rewrite/ast_helpers/ast_modifier.py +17 -4
- mindspore/rewrite/ast_helpers/ast_replacer.py +1 -1
- mindspore/rewrite/ast_transformers/__init__.py +0 -1
- mindspore/rewrite/ast_transformers/flatten_recursive_stmt.py +46 -5
- mindspore/rewrite/ast_transformers/remove_return_out_of_if.py +6 -3
- mindspore/rewrite/common/__init__.py +2 -0
- mindspore/rewrite/common/event.py +1 -1
- mindspore/rewrite/common/observable.py +1 -1
- mindspore/rewrite/common/observer.py +1 -1
- mindspore/rewrite/common/rewrite_elog.py +35 -0
- mindspore/rewrite/namer.py +2 -2
- mindspore/rewrite/namespace.py +14 -4
- mindspore/rewrite/node.py +161 -13
- mindspore/rewrite/parser.py +0 -1
- mindspore/rewrite/parser_register.py +0 -1
- mindspore/rewrite/parsers/arguments_parser.py +3 -2
- mindspore/rewrite/parsers/assign_parser.py +267 -67
- mindspore/rewrite/parsers/attribute_parser.py +56 -0
- mindspore/rewrite/parsers/class_def_parser.py +191 -108
- mindspore/rewrite/parsers/constant_parser.py +101 -0
- mindspore/rewrite/parsers/container_parser.py +88 -0
- mindspore/rewrite/parsers/for_parser.py +28 -15
- mindspore/rewrite/parsers/function_def_parser.py +21 -5
- mindspore/rewrite/parsers/if_parser.py +11 -28
- mindspore/rewrite/parsers/module_parser.py +9 -6
- mindspore/rewrite/parsers/return_parser.py +3 -2
- mindspore/rewrite/sparsify/__init__.py +0 -0
- mindspore/rewrite/sparsify/sparse_transformer.py +448 -0
- mindspore/rewrite/sparsify/sparsify.py +109 -0
- mindspore/rewrite/sparsify/utils.py +173 -0
- mindspore/rewrite/symbol_tree.py +322 -109
- mindspore/rewrite/symbol_tree_builder.py +45 -8
- mindspore/rewrite/symbol_tree_dumper.py +0 -1
- mindspore/rewrite/topological_manager.py +1 -2
- mindspore/run_check/_check_version.py +209 -112
- mindspore/run_check/run_check.py +2 -1
- mindspore/tbbmalloc.dll +0 -0
- mindspore/tinyxml2.dll +0 -0
- mindspore/train/__init__.py +6 -4
- mindspore/train/_utils.py +28 -5
- mindspore/train/amp.py +321 -50
- mindspore/train/callback/__init__.py +3 -1
- mindspore/train/callback/_backup_and_restore.py +120 -0
- mindspore/train/callback/_callback.py +8 -8
- mindspore/train/callback/_checkpoint.py +12 -9
- mindspore/train/callback/_early_stop.py +13 -7
- mindspore/train/callback/_history.py +8 -8
- mindspore/train/callback/_lambda_callback.py +6 -6
- mindspore/train/callback/_landscape.py +36 -38
- mindspore/train/callback/_loss_monitor.py +12 -6
- mindspore/train/callback/_lr_scheduler_callback.py +2 -4
- mindspore/train/callback/_on_request_exit.py +212 -0
- mindspore/train/callback/_reduce_lr_on_plateau.py +13 -7
- mindspore/train/callback/_summary_collector.py +27 -19
- mindspore/train/callback/_time_monitor.py +13 -7
- mindspore/train/checkpoint_pb2.py +68 -8
- mindspore/train/data_sink.py +122 -33
- mindspore/train/dataset_helper.py +28 -87
- mindspore/train/loss_scale_manager.py +4 -7
- mindspore/{nn → train}/metrics/__init__.py +20 -20
- mindspore/{nn → train}/metrics/accuracy.py +12 -10
- mindspore/{nn → train}/metrics/auc.py +4 -4
- mindspore/{nn → train}/metrics/bleu_score.py +4 -4
- mindspore/{nn → train}/metrics/confusion_matrix.py +10 -8
- mindspore/{nn → train}/metrics/cosine_similarity.py +4 -4
- mindspore/{nn → train}/metrics/dice.py +6 -5
- mindspore/{nn → train}/metrics/error.py +7 -5
- mindspore/{nn → train}/metrics/fbeta.py +9 -7
- mindspore/{nn → train}/metrics/hausdorff_distance.py +8 -6
- mindspore/{nn → train}/metrics/loss.py +4 -3
- mindspore/{nn → train}/metrics/mean_surface_distance.py +6 -5
- mindspore/{nn → train}/metrics/metric.py +6 -5
- mindspore/{nn → train}/metrics/occlusion_sensitivity.py +4 -3
- mindspore/{nn → train}/metrics/perplexity.py +5 -4
- mindspore/{nn → train}/metrics/precision.py +5 -4
- mindspore/{nn → train}/metrics/recall.py +5 -4
- mindspore/{nn → train}/metrics/roc.py +7 -6
- mindspore/{nn → train}/metrics/root_mean_square_surface_distance.py +6 -5
- mindspore/{nn → train}/metrics/topk.py +7 -5
- mindspore/train/mind_ir_pb2.py +339 -32
- mindspore/train/model.py +113 -84
- mindspore/train/serialization.py +547 -167
- mindspore/train/summary/_summary_adapter.py +1 -1
- mindspore/train/summary/summary_record.py +43 -12
- mindspore/train/train_thor/convert_utils.py +7 -1
- mindspore/train/train_thor/dataset_helper.py +3 -3
- mindspore/train/train_thor/model_thor.py +0 -4
- mindspore/turbojpeg.dll +0 -0
- mindspore/vcmeta.dll +0 -0
- mindspore/vcruntime140.dll +0 -0
- mindspore/vcruntime140_1.dll +0 -0
- mindspore/version.py +1 -1
- {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/METADATA +4 -3
- {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/RECORD +901 -660
- mindspore/compression/common/constant.py +0 -124
- mindspore/compression/export/__init__.py +0 -19
- mindspore/compression/export/quant_export.py +0 -514
- mindspore/compression/quant/qat.py +0 -636
- mindspore/compression/quant/quant_utils.py +0 -462
- mindspore/compression/quant/quantizer.py +0 -68
- mindspore/libatomic-1.dll +0 -0
- mindspore/libgcc_s_seh-1.dll +0 -0
- mindspore/libgfortran-4.dll +0 -0
- mindspore/libgomp-1.dll +0 -0
- mindspore/libjpeg-62.dll +0 -0
- mindspore/libmindspore.dll +0 -0
- mindspore/libmindspore_common.dll +0 -0
- mindspore/libmindspore_core.dll +0 -0
- mindspore/libmindspore_glog.dll +0 -0
- mindspore/libnnacl.dll +0 -0
- mindspore/libopencv_core452.dll +0 -0
- mindspore/libopencv_imgcodecs452.dll +0 -0
- mindspore/libopencv_imgproc452.dll +0 -0
- mindspore/libquadmath-0.dll +0 -0
- mindspore/libsqlite3.dll +0 -0
- mindspore/libssp-0.dll +0 -0
- mindspore/libstdc++-6.dll +0 -0
- mindspore/libtinyxml2.dll +0 -0
- mindspore/libturbojpeg.dll +0 -0
- mindspore/libwinpthread-1.dll +0 -0
- mindspore/nn/layer/quant.py +0 -1868
- mindspore/nn/layer/rnn_utils.py +0 -90
- mindspore/nn/probability/dpn/__init__.py +0 -22
- mindspore/nn/probability/dpn/vae/__init__.py +0 -25
- mindspore/nn/probability/dpn/vae/cvae.py +0 -138
- mindspore/nn/probability/dpn/vae/vae.py +0 -122
- mindspore/nn/probability/infer/__init__.py +0 -22
- mindspore/nn/probability/infer/variational/elbo.py +0 -70
- mindspore/nn/probability/infer/variational/svi.py +0 -84
- mindspore/nn/probability/toolbox/__init__.py +0 -22
- mindspore/nn/probability/toolbox/anomaly_detection.py +0 -99
- mindspore/nn/probability/toolbox/uncertainty_evaluation.py +0 -363
- mindspore/nn/probability/transforms/__init__.py +0 -22
- mindspore/nn/probability/transforms/transform_bnn.py +0 -262
- mindspore/nn/probability/zhusuan/__init__.py +0 -18
- mindspore/nn/probability/zhusuan/framework/__init__.py +0 -18
- mindspore/nn/probability/zhusuan/framework/bn.py +0 -95
- mindspore/nn/probability/zhusuan/variational/__init__.py +0 -18
- mindspore/nn/probability/zhusuan/variational/elbo.py +0 -46
- mindspore/ops/_op_impl/tbe/bias_add_grad_ds.py +0 -52
- mindspore/ops/_op_impl/tbe/scatter_nd_add_ds.py +0 -43
- mindspore/ops/bprop_mindir/AssignAdd_bprop.mindir +0 -20
- mindspore/ops/bprop_mindir/Identity_bprop.mindir +0 -9
- mindspore/ops/bprop_mindir/LogicalOr_bprop.mindir +0 -20
- mindspore/ops/bprop_mindir/ReLU_bprop.mindir +0 -16
- mindspore/ops/bprop_mindir/UpdateState_bprop.mindir +0 -17
- mindspore/ops/bprop_mindir/stop_gradient_bprop.mindir +0 -12
- mindspore/ops/composite/array_ops.py +0 -210
- mindspore/ops/composite/clip_ops.py +0 -238
- mindspore/ops/composite/random_ops.py +0 -426
- mindspore/ops/composite/vmap_ops.py +0 -38
- mindspore/ops/operations/sponge_ops.py +0 -3531
- mindspore/ops/operations/sponge_update_ops.py +0 -2546
- mindspore/parallel/nn/__init__.py +0 -42
- mindspore/parallel/nn/loss.py +0 -22
- mindspore/parallel/nn/moe.py +0 -21
- mindspore/parallel/nn/op_parallel_config.py +0 -22
- mindspore/parallel/nn/transformer.py +0 -31
- mindspore/run_check/_check_deps_version.py +0 -84
- {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/WHEEL +0 -0
- {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/entry_points.txt +0 -0
- {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/top_level.txt +0 -0
mindspore/dataset/text/utils.py
CHANGED
|
@@ -36,16 +36,25 @@ class CharNGram(cde.CharNGram):
|
|
|
36
36
|
@check_from_file_vectors
|
|
37
37
|
def from_file(cls, file_path, max_vectors=None):
|
|
38
38
|
"""
|
|
39
|
-
Build a CharNGram vector from a file.
|
|
39
|
+
Build a `CharNGram` vector from a file.
|
|
40
40
|
|
|
41
41
|
Args:
|
|
42
|
-
file_path (str): Path of the file that contains the CharNGram vectors.
|
|
42
|
+
file_path (str): Path of the file that contains the `CharNGram` vectors.
|
|
43
43
|
max_vectors (int, optional): This can be used to limit the number of pre-trained vectors loaded.
|
|
44
44
|
Most pre-trained vector sets are sorted in the descending order of word frequency. Thus, in
|
|
45
|
-
situations where the entire set doesn
|
|
46
|
-
passing max_vectors can limit the size of the loaded set
|
|
45
|
+
situations where the entire set doesn't fit in memory, or is not needed for another reason,
|
|
46
|
+
passing `max_vectors` can limit the size of the loaded set. Default: None, no limit.
|
|
47
|
+
|
|
48
|
+
Returns:
|
|
49
|
+
CharNGram, CharNGram vector build from a file.
|
|
50
|
+
|
|
51
|
+
Raises:
|
|
52
|
+
RuntimeError: If `file_path` contains invalid data.
|
|
53
|
+
ValueError: If `max_vectors` is invalid.
|
|
54
|
+
TypeError: If `max_vectors` is not type of integer.
|
|
47
55
|
|
|
48
56
|
Examples:
|
|
57
|
+
>>> import mindspore.dataset.text as text
|
|
49
58
|
>>> char_n_gram = text.CharNGram.from_file("/path/to/char_n_gram/file", max_vectors=None)
|
|
50
59
|
"""
|
|
51
60
|
|
|
@@ -66,13 +75,22 @@ class FastText(cde.FastText):
|
|
|
66
75
|
|
|
67
76
|
Args:
|
|
68
77
|
file_path (str): Path of the file that contains the vectors. The shuffix of pre-trained vector sets
|
|
69
|
-
must be `*.vec
|
|
78
|
+
must be `*.vec` .
|
|
70
79
|
max_vectors (int, optional): This can be used to limit the number of pre-trained vectors loaded.
|
|
71
80
|
Most pre-trained vector sets are sorted in the descending order of word frequency. Thus, in
|
|
72
|
-
situations where the entire set doesn
|
|
73
|
-
passing max_vectors can limit the size of the loaded set
|
|
81
|
+
situations where the entire set doesn't fit in memory, or is not needed for another reason,
|
|
82
|
+
passing `max_vectors` can limit the size of the loaded set. Default: None, no limit.
|
|
83
|
+
|
|
84
|
+
Returns:
|
|
85
|
+
FastText, FastText vector build from a file.
|
|
86
|
+
|
|
87
|
+
Raises:
|
|
88
|
+
RuntimeError: If `file_path` contains invalid data.
|
|
89
|
+
ValueError: If `max_vectors` is invalid.
|
|
90
|
+
TypeError: If `max_vectors` is not type of integer.
|
|
74
91
|
|
|
75
92
|
Examples:
|
|
93
|
+
>>> import mindspore.dataset.text as text
|
|
76
94
|
>>> fast_text = text.FastText.from_file("/path/to/fast_text/file", max_vectors=None)
|
|
77
95
|
"""
|
|
78
96
|
|
|
@@ -93,13 +111,22 @@ class GloVe(cde.GloVe):
|
|
|
93
111
|
|
|
94
112
|
Args:
|
|
95
113
|
file_path (str): Path of the file that contains the vectors. The format of pre-trained vector sets
|
|
96
|
-
must be `glove.6B.*.txt
|
|
114
|
+
must be `glove.6B.*.txt` .
|
|
97
115
|
max_vectors (int, optional): This can be used to limit the number of pre-trained vectors loaded.
|
|
98
116
|
Most pre-trained vector sets are sorted in the descending order of word frequency. Thus, in
|
|
99
|
-
situations where the entire set doesn
|
|
100
|
-
passing max_vectors can limit the size of the loaded set
|
|
117
|
+
situations where the entire set doesn't fit in memory, or is not needed for another reason,
|
|
118
|
+
passing `max_vectors` can limit the size of the loaded set. Default: None, no limit.
|
|
119
|
+
|
|
120
|
+
Returns:
|
|
121
|
+
GloVe, GloVe vector build from a file.
|
|
122
|
+
|
|
123
|
+
Raises:
|
|
124
|
+
RuntimeError: If `file_path` contains invalid data.
|
|
125
|
+
ValueError: If `max_vectors` is invalid.
|
|
126
|
+
TypeError: If `max_vectors` is not type of integer.
|
|
101
127
|
|
|
102
128
|
Examples:
|
|
129
|
+
>>> import mindspore.dataset.text as text
|
|
103
130
|
>>> glove = text.GloVe.from_file("/path/to/glove/file", max_vectors=None)
|
|
104
131
|
"""
|
|
105
132
|
|
|
@@ -109,7 +136,7 @@ class GloVe(cde.GloVe):
|
|
|
109
136
|
|
|
110
137
|
class JiebaMode(IntEnum):
|
|
111
138
|
"""
|
|
112
|
-
An enumeration for :class:`mindspore.dataset.text.JiebaTokenizer
|
|
139
|
+
An enumeration for :class:`mindspore.dataset.text.JiebaTokenizer` .
|
|
113
140
|
|
|
114
141
|
Possible enumeration values are: JiebaMode.MIX, JiebaMode.MP, JiebaMode.HMM.
|
|
115
142
|
|
|
@@ -211,16 +238,18 @@ class SentencePieceVocab:
|
|
|
211
238
|
SentencePieceVocab, vocab built from the dataset.
|
|
212
239
|
|
|
213
240
|
Examples:
|
|
214
|
-
>>>
|
|
241
|
+
>>> import mindspore.dataset as ds
|
|
242
|
+
>>> from mindspore.dataset.text import SentencePieceVocab, SentencePieceModel
|
|
215
243
|
>>> dataset = ds.TextFileDataset("/path/to/sentence/piece/vocab/file", shuffle=False)
|
|
216
|
-
>>> vocab =
|
|
217
|
-
...
|
|
244
|
+
>>> vocab = SentencePieceVocab.from_dataset(dataset, ["text"], 5000, 0.9995,
|
|
245
|
+
... SentencePieceModel.UNIGRAM, {})
|
|
218
246
|
"""
|
|
219
247
|
|
|
220
248
|
sentence_piece_vocab = cls()
|
|
221
|
-
|
|
222
|
-
|
|
223
|
-
|
|
249
|
+
# pylint: disable=protected-access
|
|
250
|
+
sentence_piece_vocab.c_sentence_piece_vocab = dataset._build_sentencepiece_vocab(col_names, vocab_size,
|
|
251
|
+
character_coverage,
|
|
252
|
+
model_type, params)
|
|
224
253
|
return sentence_piece_vocab
|
|
225
254
|
|
|
226
255
|
@classmethod
|
|
@@ -253,9 +282,9 @@ class SentencePieceVocab:
|
|
|
253
282
|
SentencePieceVocab, vocab built from the file.
|
|
254
283
|
|
|
255
284
|
Examples:
|
|
256
|
-
>>> from mindspore.dataset.text import SentencePieceModel
|
|
257
|
-
>>> vocab =
|
|
258
|
-
...
|
|
285
|
+
>>> from mindspore.dataset.text import SentencePieceVocab, SentencePieceModel
|
|
286
|
+
>>> vocab = SentencePieceVocab.from_file(["/path/to/sentence/piece/vocab/file"], 5000, 0.9995,
|
|
287
|
+
... SentencePieceModel.UNIGRAM, {})
|
|
259
288
|
"""
|
|
260
289
|
|
|
261
290
|
sentence_piece_vocab = cls()
|
|
@@ -275,10 +304,10 @@ class SentencePieceVocab:
|
|
|
275
304
|
filename (str): The name of the file.
|
|
276
305
|
|
|
277
306
|
Examples:
|
|
278
|
-
>>> from mindspore.dataset.text import SentencePieceModel
|
|
279
|
-
>>> vocab =
|
|
280
|
-
...
|
|
281
|
-
>>>
|
|
307
|
+
>>> from mindspore.dataset.text import SentencePieceVocab, SentencePieceModel
|
|
308
|
+
>>> vocab = SentencePieceVocab.from_file(["/path/to/sentence/piece/vocab/file"], 5000, 0.9995,
|
|
309
|
+
... SentencePieceModel.UNIGRAM, {})
|
|
310
|
+
>>> SentencePieceVocab.save_model(vocab, "./", "m.model")
|
|
282
311
|
"""
|
|
283
312
|
|
|
284
313
|
cde.SentencePieceVocab.save_model(vocab.c_sentence_piece_vocab, path, filename)
|
|
@@ -286,7 +315,7 @@ class SentencePieceVocab:
|
|
|
286
315
|
|
|
287
316
|
class SPieceTokenizerLoadType(IntEnum):
|
|
288
317
|
"""
|
|
289
|
-
An enumeration for loading type of :class:`mindspore.dataset.text.SentencePieceTokenizer
|
|
318
|
+
An enumeration for loading type of :class:`mindspore.dataset.text.SentencePieceTokenizer` .
|
|
290
319
|
|
|
291
320
|
Possible enumeration values are: SPieceTokenizerLoadType.FILE, SPieceTokenizerLoadType.MODEL.
|
|
292
321
|
|
|
@@ -300,7 +329,7 @@ class SPieceTokenizerLoadType(IntEnum):
|
|
|
300
329
|
|
|
301
330
|
class SPieceTokenizerOutType(IntEnum):
|
|
302
331
|
"""
|
|
303
|
-
An enumeration for :class:`mindspore.dataset.text.SentencePieceTokenizer
|
|
332
|
+
An enumeration for :class:`mindspore.dataset.text.SentencePieceTokenizer` .
|
|
304
333
|
|
|
305
334
|
Possible enumeration values are: SPieceTokenizerOutType.STRING, SPieceTokenizerOutType.INT.
|
|
306
335
|
|
|
@@ -327,10 +356,19 @@ class Vectors(cde.Vectors):
|
|
|
327
356
|
file_path (str): Path of the file that contains the vectors.
|
|
328
357
|
max_vectors (int, optional): This can be used to limit the number of pre-trained vectors loaded.
|
|
329
358
|
Most pre-trained vector sets are sorted in the descending order of word frequency. Thus, in
|
|
330
|
-
situations where the entire set doesn
|
|
331
|
-
passing max_vectors can limit the size of the loaded set
|
|
359
|
+
situations where the entire set doesn't fit in memory, or is not needed for another reason,
|
|
360
|
+
passing `max_vectors` can limit the size of the loaded set. Default: None, no limit.
|
|
361
|
+
|
|
362
|
+
Returns:
|
|
363
|
+
Vectors, Vectors build from a file.
|
|
364
|
+
|
|
365
|
+
Raises:
|
|
366
|
+
RuntimeError: If `file_path` contains invalid data.
|
|
367
|
+
ValueError: If `max_vectors` is invalid.
|
|
368
|
+
TypeError: If `max_vectors` is not type of integer.
|
|
332
369
|
|
|
333
370
|
Examples:
|
|
371
|
+
>>> import mindspore.dataset.text as text
|
|
334
372
|
>>> vector = text.Vectors.from_file("/path/to/vectors/file", max_vectors=None)
|
|
335
373
|
"""
|
|
336
374
|
|
|
@@ -362,25 +400,27 @@ class Vocab:
|
|
|
362
400
|
Args:
|
|
363
401
|
dataset (Dataset): dataset to build vocab from.
|
|
364
402
|
columns (list[str], optional): column names to get words from. It can be a list of column names.
|
|
365
|
-
|
|
403
|
+
Default: None.
|
|
366
404
|
freq_range (tuple, optional): A tuple of integers (min_frequency, max_frequency). Words within the frequency
|
|
367
405
|
range would be kept. 0 <= min_frequency <= max_frequency <= total_words. min_frequency=0 is the same as
|
|
368
406
|
min_frequency=1. max_frequency > total_words is the same as max_frequency = total_words.
|
|
369
|
-
min_frequency/max_frequency can be None, which corresponds to 0/total_words separately
|
|
370
|
-
|
|
407
|
+
min_frequency/max_frequency can be None, which corresponds to 0/total_words separately.
|
|
408
|
+
Default: None, all words are included.
|
|
371
409
|
top_k (int, optional): top_k is greater than 0. Number of words to be built into vocab. top_k means most
|
|
372
|
-
frequent words are taken. top_k is taken after freq_range. If not enough top_k, all words will be taken
|
|
373
|
-
|
|
410
|
+
frequent words are taken. top_k is taken after freq_range. If not enough top_k, all words will be taken.
|
|
411
|
+
Default: None, all words are included.
|
|
374
412
|
special_tokens (list, optional): A list of strings, each one is a special token. For example
|
|
375
|
-
special_tokens=["<pad>","<unk>"]
|
|
413
|
+
special_tokens=["<pad>","<unk>"]. Default: None, no special tokens will be added.
|
|
376
414
|
special_first (bool, optional): Whether special_tokens will be prepended/appended to vocab. If
|
|
377
|
-
special_tokens is specified and special_first is set to True, special_tokens will be prepended
|
|
378
|
-
|
|
415
|
+
special_tokens is specified and special_first is set to True, special_tokens will be prepended.
|
|
416
|
+
Default: True.
|
|
379
417
|
|
|
380
418
|
Returns:
|
|
381
419
|
Vocab, Vocab object built from the dataset.
|
|
382
420
|
|
|
383
421
|
Examples:
|
|
422
|
+
>>> import mindspore.dataset as ds
|
|
423
|
+
>>> import mindspore.dataset.text as text
|
|
384
424
|
>>> dataset = ds.TextFileDataset("/path/to/sentence/piece/vocab/file", shuffle=False)
|
|
385
425
|
>>> vocab = text.Vocab.from_dataset(dataset, "text", freq_range=None, top_k=None,
|
|
386
426
|
... special_tokens=["<pad>", "<unk>"],
|
|
@@ -389,7 +429,8 @@ class Vocab:
|
|
|
389
429
|
"""
|
|
390
430
|
|
|
391
431
|
vocab = cls()
|
|
392
|
-
|
|
432
|
+
# pylint: disable=protected-access
|
|
433
|
+
vocab.c_vocab = dataset._build_vocab(columns, freq_range, top_k, special_tokens, special_first)
|
|
393
434
|
return vocab
|
|
394
435
|
|
|
395
436
|
@classmethod
|
|
@@ -401,14 +442,15 @@ class Vocab:
|
|
|
401
442
|
Args:
|
|
402
443
|
word_list (list): A list of string where each element is a word of type string.
|
|
403
444
|
special_tokens (list, optional): A list of strings, each one is a special token. For example
|
|
404
|
-
special_tokens=["<pad>","<unk>"]
|
|
445
|
+
special_tokens=["<pad>","<unk>"]. Default: None, no special tokens will be added.
|
|
405
446
|
special_first (bool, optional): Whether special_tokens is prepended or appended to vocab. If special_tokens
|
|
406
|
-
is specified and special_first is set to True, special_tokens will be prepended
|
|
447
|
+
is specified and special_first is set to True, special_tokens will be prepended. Default: True.
|
|
407
448
|
|
|
408
449
|
Returns:
|
|
409
450
|
Vocab, Vocab object built from the list.
|
|
410
451
|
|
|
411
452
|
Examples:
|
|
453
|
+
>>> import mindspore.dataset.text as text
|
|
412
454
|
>>> vocab = text.Vocab.from_list(["w1", "w2", "w3"], special_tokens=["<unk>"], special_first=True)
|
|
413
455
|
"""
|
|
414
456
|
|
|
@@ -427,18 +469,19 @@ class Vocab:
|
|
|
427
469
|
Args:
|
|
428
470
|
file_path (str): Path to the file which contains the vocab list.
|
|
429
471
|
delimiter (str, optional): A delimiter to break up each line in file, the first element is taken to be
|
|
430
|
-
the word
|
|
431
|
-
vocab_size (int, optional): Number of words to read from file_path
|
|
472
|
+
the word. Default: '', the whole line will be treated as a word.
|
|
473
|
+
vocab_size (int, optional): Number of words to read from file_path. Default: None, all words are taken.
|
|
432
474
|
special_tokens (list, optional): A list of strings, each one is a special token. For example
|
|
433
|
-
special_tokens=["<pad>","<unk>"]
|
|
475
|
+
special_tokens=["<pad>","<unk>"]. Default: None, no special tokens will be added.
|
|
434
476
|
special_first (bool, optional): Whether special_tokens will be prepended/appended to vocab,
|
|
435
477
|
If special_tokens is specified and special_first is set to True,
|
|
436
|
-
special_tokens will be prepended
|
|
478
|
+
special_tokens will be prepended. Default: True.
|
|
437
479
|
|
|
438
480
|
Returns:
|
|
439
481
|
Vocab, Vocab object built from the file.
|
|
440
482
|
|
|
441
483
|
Examples:
|
|
484
|
+
>>> import mindspore.dataset.text as text
|
|
442
485
|
>>> # Assume vocab file contains the following content:
|
|
443
486
|
>>> # --- begin of file ---
|
|
444
487
|
>>> # apple,apple2
|
|
@@ -476,6 +519,7 @@ class Vocab:
|
|
|
476
519
|
Vocab, Vocab object built from the dict.
|
|
477
520
|
|
|
478
521
|
Examples:
|
|
522
|
+
>>> import mindspore.dataset.text as text
|
|
479
523
|
>>> vocab = text.Vocab.from_dict({"home": 3, "behind": 2, "the": 4, "world": 5, "<unk>": 6})
|
|
480
524
|
"""
|
|
481
525
|
|
|
@@ -491,6 +535,7 @@ class Vocab:
|
|
|
491
535
|
A vocabulary consisting of word and id pairs.
|
|
492
536
|
|
|
493
537
|
Examples:
|
|
538
|
+
>>> import mindspore.dataset.text as text
|
|
494
539
|
>>> vocab = text.Vocab.from_list(["word_1", "word_2", "word_3", "word_4"])
|
|
495
540
|
>>> vocabory_dict = vocab.vocab()
|
|
496
541
|
"""
|
|
@@ -510,6 +555,7 @@ class Vocab:
|
|
|
510
555
|
The token id or list of token ids.
|
|
511
556
|
|
|
512
557
|
Examples:
|
|
558
|
+
>>> import mindspore.dataset.text as text
|
|
513
559
|
>>> vocab = text.Vocab.from_list(["w1", "w2", "w3"], special_tokens=["<unk>"], special_first=True)
|
|
514
560
|
>>> ids = vocab.tokens_to_ids(["w1", "w3"])
|
|
515
561
|
"""
|
|
@@ -533,6 +579,7 @@ class Vocab:
|
|
|
533
579
|
The decoded token(s).
|
|
534
580
|
|
|
535
581
|
Examples:
|
|
582
|
+
>>> import mindspore.dataset.text as text
|
|
536
583
|
>>> vocab = text.Vocab.from_list(["w1", "w2", "w3"], special_tokens=["<unk>"], special_first=True)
|
|
537
584
|
>>> token = vocab.ids_to_tokens(0)
|
|
538
585
|
"""
|
|
@@ -546,17 +593,18 @@ class Vocab:
|
|
|
546
593
|
|
|
547
594
|
def to_bytes(array, encoding='utf8'):
|
|
548
595
|
"""
|
|
549
|
-
Convert NumPy array of `str` to array of `bytes` by encoding each element based on charset `encoding
|
|
596
|
+
Convert NumPy array of `str` to array of `bytes` by encoding each element based on charset `encoding` .
|
|
550
597
|
|
|
551
598
|
Args:
|
|
552
599
|
array (numpy.ndarray): Array of `str` type representing strings.
|
|
553
|
-
encoding (str): Indicating the charset for encoding
|
|
600
|
+
encoding (str): Indicating the charset for encoding. Default: 'utf8'.
|
|
554
601
|
|
|
555
602
|
Returns:
|
|
556
|
-
numpy.ndarray, NumPy array of `bytes
|
|
603
|
+
numpy.ndarray, NumPy array of `bytes` .
|
|
557
604
|
|
|
558
605
|
Examples:
|
|
559
606
|
>>> import numpy as np
|
|
607
|
+
>>> import mindspore.dataset as ds
|
|
560
608
|
>>>
|
|
561
609
|
>>> data = np.array([["1", "2", "3"]], dtype=np.str_)
|
|
562
610
|
>>> dataset = ds.NumpySlicesDataset(data, column_names=["text"])
|
|
@@ -572,17 +620,18 @@ def to_bytes(array, encoding='utf8'):
|
|
|
572
620
|
|
|
573
621
|
def to_str(array, encoding='utf8'):
|
|
574
622
|
"""
|
|
575
|
-
Convert NumPy array of `bytes` to array of `str` by decoding each element based on charset `encoding
|
|
623
|
+
Convert NumPy array of `bytes` to array of `str` by decoding each element based on charset `encoding` .
|
|
576
624
|
|
|
577
625
|
Args:
|
|
578
626
|
array (numpy.ndarray): Array of `bytes` type representing strings.
|
|
579
|
-
encoding (str): Indicating the charset for decoding
|
|
627
|
+
encoding (str): Indicating the charset for decoding. Default: 'utf8'.
|
|
580
628
|
|
|
581
629
|
Returns:
|
|
582
|
-
numpy.ndarray, NumPy array of `str
|
|
630
|
+
numpy.ndarray, NumPy array of `str` .
|
|
583
631
|
|
|
584
632
|
Examples:
|
|
585
633
|
>>> import numpy as np
|
|
634
|
+
>>> import mindspore.dataset as ds
|
|
586
635
|
>>>
|
|
587
636
|
>>> data = np.array([["1", "2", "3"]], dtype=np.bytes_)
|
|
588
637
|
>>> dataset = ds.NumpySlicesDataset(data, column_names=["text"])
|
|
@@ -27,6 +27,19 @@ from ..core.validator_helpers import parse_user_args, type_check, type_check_lis
|
|
|
27
27
|
INT32_MAX, check_value, check_positive, check_pos_int32, check_filename, check_non_negative_int32
|
|
28
28
|
|
|
29
29
|
|
|
30
|
+
def check_add_token(method):
|
|
31
|
+
"""Wrapper method to check the parameters of add token."""
|
|
32
|
+
|
|
33
|
+
@wraps(method)
|
|
34
|
+
def new_method(self, *args, **kwargs):
|
|
35
|
+
[token, begin], _ = parse_user_args(method, *args, **kwargs)
|
|
36
|
+
type_check(token, (str,), "token")
|
|
37
|
+
type_check(begin, (bool,), "begin")
|
|
38
|
+
return method(self, *args, **kwargs)
|
|
39
|
+
|
|
40
|
+
return new_method
|
|
41
|
+
|
|
42
|
+
|
|
30
43
|
def check_unique_list_of_words(words, arg_name):
|
|
31
44
|
"""Check that words is a list and each element is a str without any duplication"""
|
|
32
45
|
|
|
@@ -443,6 +456,18 @@ def check_ngram(method):
|
|
|
443
456
|
return new_method
|
|
444
457
|
|
|
445
458
|
|
|
459
|
+
def check_truncate(method):
|
|
460
|
+
"""Wrapper method to check the parameters of number of truncate."""
|
|
461
|
+
|
|
462
|
+
@wraps(method)
|
|
463
|
+
def new_method(self, *args, **kwargs):
|
|
464
|
+
[max_seq_len], _ = parse_user_args(method, *args, **kwargs)
|
|
465
|
+
check_pos_int32(max_seq_len, "max_seq_len")
|
|
466
|
+
return method(self, *args, **kwargs)
|
|
467
|
+
|
|
468
|
+
return new_method
|
|
469
|
+
|
|
470
|
+
|
|
446
471
|
def check_pair_truncate(method):
|
|
447
472
|
"""Wrapper method to check the parameters of number of pair truncate."""
|
|
448
473
|
|
|
@@ -30,13 +30,15 @@ Note: Legacy c_transforms and py_transforms are deprecated but can still be impo
|
|
|
30
30
|
from mindspore.dataset.transforms import c_transforms
|
|
31
31
|
from mindspore.dataset.transforms import py_transforms
|
|
32
32
|
|
|
33
|
-
See `Common
|
|
34
|
-
<https://www.mindspore.cn/tutorials/en/
|
|
33
|
+
See `Common Transforms
|
|
34
|
+
<https://www.mindspore.cn/tutorials/en/r2.0/beginner/transforms.html#common-transforms>`_ tutorial for more details.
|
|
35
35
|
|
|
36
36
|
Descriptions of common data processing terms are as follows:
|
|
37
37
|
|
|
38
38
|
- TensorOperation, the base class of all data processing operations implemented in C++.
|
|
39
39
|
- PyTensorOperation, the base class of all data processing operations implemented in Python.
|
|
40
|
+
|
|
41
|
+
Note: In eager mode, non-NumPy input is implicitly converted to NumPy format and sent to MindSpore.
|
|
40
42
|
"""
|
|
41
43
|
from .. import vision
|
|
42
44
|
from . import c_transforms
|
|
@@ -142,7 +142,7 @@ class TypeCast(TensorOperation):
|
|
|
142
142
|
TypeError: If `data_type` is not of type bool, int, float or string.
|
|
143
143
|
|
|
144
144
|
Supported Platforms:
|
|
145
|
-
``
|
|
145
|
+
``Ascend`` ``GPU`` ``CPU``
|
|
146
146
|
|
|
147
147
|
Examples:
|
|
148
148
|
>>> import numpy as np
|
|
@@ -211,10 +211,10 @@ class Slice(TensorOperation):
|
|
|
211
211
|
`slice <https://docs.python.org/3.7/library/functions.html?highlight=slice#slice>`_ object along the
|
|
212
212
|
first dimension. Similar to start:stop:step.
|
|
213
213
|
4. :py:obj:`None`: Slice the whole dimension. Similar to :py:obj:`[:]` in Python indexing.
|
|
214
|
-
5. :py:obj:`Ellipsis`: Slice the whole dimension, same result with `None
|
|
214
|
+
5. :py:obj:`Ellipsis`: Slice the whole dimension, same result with `None` .
|
|
215
215
|
|
|
216
216
|
Raises:
|
|
217
|
-
TypeError: If `slices` is not of type int, list[int], :py:obj:`slice
|
|
217
|
+
TypeError: If `slices` is not of type int, list[int], :py:obj:`slice` , :py:obj:`None` or :py:obj:`Ellipsis` .
|
|
218
218
|
|
|
219
219
|
Supported Platforms:
|
|
220
220
|
``CPU``
|
|
@@ -286,7 +286,7 @@ class Mask(TensorOperation):
|
|
|
286
286
|
operator (Relational): relational operators, it can be any of [Relational.EQ, Relational.NE, Relational.LT,
|
|
287
287
|
Relational.GT, Relational.LE, Relational.GE], take Relational.EQ as example, EQ refers to equal.
|
|
288
288
|
constant (Union[str, int, float, bool]): Constant to be compared to.
|
|
289
|
-
dtype (mindspore.dtype, optional): Type of the generated mask. Default:
|
|
289
|
+
dtype (mindspore.dtype, optional): Type of the generated mask. Default: mstype.bool_.
|
|
290
290
|
|
|
291
291
|
Raises:
|
|
292
292
|
TypeError: `operator` is not of type Relational.
|
|
@@ -375,10 +375,10 @@ class Concatenate(TensorOperation):
|
|
|
375
375
|
Tensor operation that concatenates all columns into a single tensor.
|
|
376
376
|
|
|
377
377
|
Args:
|
|
378
|
-
axis (int, optional): Concatenate the tensors along given axis
|
|
379
|
-
prepend (numpy.array, optional): NumPy array to be prepended to the already concatenated tensors
|
|
380
|
-
|
|
381
|
-
append (numpy.array, optional): NumPy array to be appended to the already concatenated tensors
|
|
378
|
+
axis (int, optional): Concatenate the tensors along given axis. Default: 0.
|
|
379
|
+
prepend (numpy.array, optional): NumPy array to be prepended to the already concatenated tensors.
|
|
380
|
+
Default: None.
|
|
381
|
+
append (numpy.array, optional): NumPy array to be appended to the already concatenated tensors. Default: None.
|
|
382
382
|
|
|
383
383
|
Raises:
|
|
384
384
|
TypeError: If `axis` is not of type int.
|
|
@@ -430,8 +430,7 @@ class Duplicate(TensorOperation):
|
|
|
430
430
|
>>> numpy_slices_dataset = ds.NumpySlicesDataset(data, ["x"])
|
|
431
431
|
>>> numpy_slices_dataset = numpy_slices_dataset.map(operations=c_transforms.Duplicate(),
|
|
432
432
|
... input_columns=["x"],
|
|
433
|
-
... output_columns=["x", "y"]
|
|
434
|
-
... column_order=["x", "y"])
|
|
433
|
+
... output_columns=["x", "y"])
|
|
435
434
|
>>> # Data after
|
|
436
435
|
>>> # | x | y |
|
|
437
436
|
>>> # +---------+---------+
|
|
@@ -477,8 +476,7 @@ class Unique(TensorOperation):
|
|
|
477
476
|
>>> dataset = ds.NumpySlicesDataset(data, ["x"])
|
|
478
477
|
>>> dataset = dataset.map(operations=c_transforms.Unique(),
|
|
479
478
|
... input_columns=["x"],
|
|
480
|
-
... output_columns=["x", "y", "z"]
|
|
481
|
-
... column_order=["x", "y", "z"])
|
|
479
|
+
... output_columns=["x", "y", "z"])
|
|
482
480
|
>>> # Data after
|
|
483
481
|
>>> # | x | y |z |
|
|
484
482
|
>>> # +---------+-----------------+---------+
|
|
@@ -536,7 +534,7 @@ class RandomApply(TensorOperation):
|
|
|
536
534
|
|
|
537
535
|
Args:
|
|
538
536
|
transforms (list): List of transformations to be applied.
|
|
539
|
-
prob (float, optional): The probability to apply the transformation list
|
|
537
|
+
prob (float, optional): The probability to apply the transformation list. Default: 0.5.
|
|
540
538
|
|
|
541
539
|
Raises:
|
|
542
540
|
TypeError: If `transforms` is not of type list.
|
|
@@ -94,7 +94,7 @@ class OneHotOp(PyTensorOperation):
|
|
|
94
94
|
num_classes (int): Number of classes of objects in dataset.
|
|
95
95
|
It should be larger than the largest label number in the dataset.
|
|
96
96
|
smoothing_rate (float, optional): Adjustable hyperparameter for label smoothing level.
|
|
97
|
-
|
|
97
|
+
Default: 0.0, means no smoothing is applied.
|
|
98
98
|
|
|
99
99
|
Raises:
|
|
100
100
|
TypeError: `num_classes` is not of type int.
|
|
@@ -260,7 +260,7 @@ class RandomApply(PyTensorOperation):
|
|
|
260
260
|
|
|
261
261
|
Args:
|
|
262
262
|
transforms (list): List of transformations to apply.
|
|
263
|
-
prob (float, optional): The probability to apply the transformation list
|
|
263
|
+
prob (float, optional): The probability to apply the transformation list. Default: 0.5.
|
|
264
264
|
|
|
265
265
|
Raises:
|
|
266
266
|
TypeError: If `transforms` is not of type list.
|
|
@@ -191,6 +191,7 @@ class FuncWrapper:
|
|
|
191
191
|
self.random = False
|
|
192
192
|
except KeyError:
|
|
193
193
|
self.random = True
|
|
194
|
+
self.logged_list_mixed_type_warning = False # Warning for list mixed type result is not logged yet
|
|
194
195
|
|
|
195
196
|
def __call__(self, *args):
|
|
196
197
|
try:
|
|
@@ -198,6 +199,15 @@ class FuncWrapper:
|
|
|
198
199
|
except Exception:
|
|
199
200
|
result = ExceptionHandler(where="in map(or batch) worker and execute Python function")
|
|
200
201
|
result.reraise()
|
|
202
|
+
|
|
203
|
+
# Check if result is list type, and mixed type warning for list type result has not been logged yet
|
|
204
|
+
if isinstance(result, list) and not self.logged_list_mixed_type_warning:
|
|
205
|
+
result0_type = type(result[0])
|
|
206
|
+
if not all((type(t) is result0_type) for t in result): # pylint: disable=unidiomatic-typecheck
|
|
207
|
+
self.logged_list_mixed_type_warning = True
|
|
208
|
+
warn_msg = "All elements in returned list are not of the same type in Python function." + \
|
|
209
|
+
" First element has type: " + str(result0_type)
|
|
210
|
+
logger.warning(warn_msg)
|
|
201
211
|
return result
|
|
202
212
|
|
|
203
213
|
def to_json(self):
|
|
@@ -346,10 +346,10 @@ class Concatenate(TensorOperation):
|
|
|
346
346
|
Tensor operation that concatenates all columns into a single tensor, only 1D tenspr is supported.
|
|
347
347
|
|
|
348
348
|
Args:
|
|
349
|
-
axis (int, optional): Concatenate the tensors along given axis
|
|
350
|
-
prepend (numpy.ndarray, optional): NumPy array to be prepended to the already concatenated tensors
|
|
351
|
-
|
|
352
|
-
append (numpy.ndarray, optional): NumPy array to be appended to the already concatenated tensors
|
|
349
|
+
axis (int, optional): Concatenate the tensors along given axis. Default: 0.
|
|
350
|
+
prepend (numpy.ndarray, optional): NumPy array to be prepended to the already concatenated tensors.
|
|
351
|
+
Default: None.
|
|
352
|
+
append (numpy.ndarray, optional): NumPy array to be appended to the already concatenated tensors. Default: None.
|
|
353
353
|
|
|
354
354
|
Raises:
|
|
355
355
|
TypeError: If `axis` is not of type int.
|
|
@@ -402,8 +402,7 @@ class Duplicate(TensorOperation):
|
|
|
402
402
|
>>> numpy_slices_dataset = ds.NumpySlicesDataset(data, ["x"])
|
|
403
403
|
>>> numpy_slices_dataset = numpy_slices_dataset.map(operations=transforms.Duplicate(),
|
|
404
404
|
... input_columns=["x"],
|
|
405
|
-
... output_columns=["x", "y"]
|
|
406
|
-
... column_order=["x", "y"])
|
|
405
|
+
... output_columns=["x", "y"])
|
|
407
406
|
>>> # Data after
|
|
408
407
|
>>> # | x | y |
|
|
409
408
|
>>> # +---------+---------+
|
|
@@ -514,7 +513,7 @@ class OneHot(TensorOperation):
|
|
|
514
513
|
num_classes (int): Number of classes of objects in dataset.
|
|
515
514
|
It should be larger than the largest label number in the dataset.
|
|
516
515
|
smoothing_rate (float, optional): Adjustable hyperparameter for label smoothing level.
|
|
517
|
-
|
|
516
|
+
Default: 0.0, means no smoothing is applied.
|
|
518
517
|
|
|
519
518
|
Raises:
|
|
520
519
|
TypeError: `num_classes` is not of type int.
|
|
@@ -630,7 +629,7 @@ class RandomApply(CompoundOperation):
|
|
|
630
629
|
|
|
631
630
|
Args:
|
|
632
631
|
transforms (list): List of transformations to be applied.
|
|
633
|
-
prob (float, optional): The probability to apply the transformation list
|
|
632
|
+
prob (float, optional): The probability to apply the transformation list. Default: 0.5.
|
|
634
633
|
|
|
635
634
|
Raises:
|
|
636
635
|
TypeError: If `transforms` is not of type list.
|
|
@@ -845,10 +844,10 @@ class Slice(TensorOperation):
|
|
|
845
844
|
`slice <https://docs.python.org/3.7/library/functions.html?highlight=slice#slice>`_ object along the
|
|
846
845
|
first dimension. Similar to start:stop:step.
|
|
847
846
|
4. :py:obj:`None`: Slice the whole dimension. Similar to :py:obj:`[:]` in Python indexing.
|
|
848
|
-
5. :py:obj:`Ellipsis`: Slice the whole dimension, same result with `None
|
|
847
|
+
5. :py:obj:`Ellipsis`: Slice the whole dimension, same result with `None` .
|
|
849
848
|
|
|
850
849
|
Raises:
|
|
851
|
-
TypeError: If `slices` is not of type int, list[int], :py:obj:`slice
|
|
850
|
+
TypeError: If `slices` is not of type int, list[int], :py:obj:`slice` , :py:obj:`None` or :py:obj:`Ellipsis` .
|
|
852
851
|
|
|
853
852
|
Supported Platforms:
|
|
854
853
|
``CPU``
|
|
@@ -890,14 +889,14 @@ class TypeCast(TensorOperation):
|
|
|
890
889
|
This operation supports running on Ascend or GPU platforms by Offload.
|
|
891
890
|
|
|
892
891
|
Args:
|
|
893
|
-
data_type (Union[mindspore.dtype, numpy.dtype]): mindspore.dtype or numpy.dtype (e.g.
|
|
892
|
+
data_type (Union[mindspore.dtype, numpy.dtype]): mindspore.dtype or numpy.dtype (e.g. `numpy.float32`)
|
|
894
893
|
to be cast to.
|
|
895
894
|
|
|
896
895
|
Raises:
|
|
897
|
-
TypeError: If `data_type` is not of MindSpore data type bool, int, float, string or type :class:`numpy.dtype
|
|
896
|
+
TypeError: If `data_type` is not of MindSpore data type bool, int, float, string or type :class:`numpy.dtype` .
|
|
898
897
|
|
|
899
898
|
Supported Platforms:
|
|
900
|
-
``
|
|
899
|
+
``Ascend`` ``GPU`` ``CPU``
|
|
901
900
|
|
|
902
901
|
Examples:
|
|
903
902
|
>>> import numpy as np
|
|
@@ -957,8 +956,7 @@ class Unique(TensorOperation):
|
|
|
957
956
|
>>> dataset = ds.NumpySlicesDataset(data, ["x"])
|
|
958
957
|
>>> dataset = dataset.map(operations=transforms.Unique(),
|
|
959
958
|
... input_columns=["x"],
|
|
960
|
-
... output_columns=["x", "y", "z"]
|
|
961
|
-
... column_order=["x", "y", "z"])
|
|
959
|
+
... output_columns=["x", "y", "z"])
|
|
962
960
|
>>> # Data after
|
|
963
961
|
>>> # | x | y |z |
|
|
964
962
|
>>> # +---------+-----------------+---------+
|
|
@@ -245,7 +245,7 @@ def check_random_transform_ops(method):
|
|
|
245
245
|
|
|
246
246
|
|
|
247
247
|
def check_transform_op_type(ind, op):
|
|
248
|
-
"""Check the
|
|
248
|
+
"""Check the operation."""
|
|
249
249
|
# c_vision.HWC2CHW error
|
|
250
250
|
# py_vision.HWC2CHW error
|
|
251
251
|
if type(op) == type: # pylint: disable=unidiomatic-typecheck
|
|
@@ -389,22 +389,22 @@ def check_type_cast(method):
|
|
|
389
389
|
|
|
390
390
|
|
|
391
391
|
def deprecated_c_transforms(substitute_name=None, substitute_module=None):
|
|
392
|
-
"""Decorator for version 1.8 deprecation warning for legacy mindspore.dataset.transforms.c_transforms
|
|
392
|
+
"""Decorator for version 1.8 deprecation warning for legacy mindspore.dataset.transforms.c_transforms operation.
|
|
393
393
|
|
|
394
394
|
Args:
|
|
395
|
-
substitute_name (str, optional): The substitute name for deprecated
|
|
396
|
-
substitute_module (str, optional): The substitute module for deprecated
|
|
395
|
+
substitute_name (str, optional): The substitute name for deprecated operation.
|
|
396
|
+
substitute_module (str, optional): The substitute module for deprecated operation.
|
|
397
397
|
"""
|
|
398
398
|
return deprecator_factory("1.8", "mindspore.dataset.transforms.c_transforms", "mindspore.dataset.transforms",
|
|
399
399
|
substitute_name, substitute_module)
|
|
400
400
|
|
|
401
401
|
|
|
402
402
|
def deprecated_py_transforms(substitute_name=None, substitute_module=None):
|
|
403
|
-
"""Decorator for version 1.8 deprecation warning for legacy mindspore.dataset.transforms.py_transforms
|
|
403
|
+
"""Decorator for version 1.8 deprecation warning for legacy mindspore.dataset.transforms.py_transforms operation.
|
|
404
404
|
|
|
405
405
|
Args:
|
|
406
|
-
substitute_name (str, optional): The substitute name for deprecated
|
|
407
|
-
substitute_module (str, optional): The substitute module for deprecated
|
|
406
|
+
substitute_name (str, optional): The substitute name for deprecated operation.
|
|
407
|
+
substitute_module (str, optional): The substitute module for deprecated operation.
|
|
408
408
|
"""
|
|
409
409
|
return deprecator_factory("1.8", "mindspore.dataset.transforms.py_transforms", "mindspore.dataset.transforms",
|
|
410
410
|
substitute_name, substitute_module)
|
|
@@ -14,5 +14,6 @@
|
|
|
14
14
|
# ==============================================================================
|
|
15
15
|
"""init file for MindData utils"""
|
|
16
16
|
from .browse_dataset import imshow_det_bbox
|
|
17
|
+
from .line_reader import LineReader
|
|
17
18
|
|
|
18
|
-
__all__ = ["imshow_det_bbox"]
|
|
19
|
+
__all__ = ["imshow_det_bbox", "LineReader"]
|