docs/api/api_python/mindspore.rst - master - envoy.istio/mindspore

docs/api/api_python/mindspore.rst
212 строк4 KB
yanglong
函数式接口支持并行文档补充
25 июн 2024, 04:52
25 июн 2024, 04:52009a247
mindspore
=========

数据表达
---------

张量
^^^^^

.. mscnautosummary::
    :toctree: mindspore

    mindspore.Tensor
    mindspore.tensor
    mindspore.COOTensor
    mindspore.CSRTensor
    mindspore.RowTensor
    mindspore.SparseTensor

参数
^^^^^

.. mscnautosummary::
    :toctree: mindspore

    mindspore.Parameter
    mindspore.ParameterTuple

数据类型
^^^^^^^^^

.. mscnautosummary::
    :toctree: mindspore

    mindspore.dtype
    mindspore.dtype_to_nptype
    mindspore.dtype_to_pytype
    mindspore.pytype_to_dtype
    mindspore.get_py_obj_dtype
    mindspore.QuantDtype
    mindspore.common.np_dtype

运行环境
---------

.. mscnautosummary::
    :toctree: mindspore

    mindspore.set_context
    mindspore.get_context
    mindspore.set_auto_parallel_context
    mindspore.get_auto_parallel_context
    mindspore.reset_auto_parallel_context
    mindspore.ParallelMode
    mindspore.set_ps_context
    mindspore.get_ps_context
    mindspore.reset_ps_context
    mindspore.set_algo_parameters
    mindspore.get_algo_parameters
    mindspore.reset_algo_parameters
    mindspore.set_offload_context
    mindspore.get_offload_context

随机种子
---------

.. mscnautosummary::
    :toctree: mindspore

    mindspore.set_seed
    mindspore.get_seed

随机状态管理
--------------

.. mscnautosummary::
    :toctree: mindspore

    mindspore.get_rng_state
    mindspore.Generator
    mindspore.initial_seed
    mindspore.manual_seed
    mindspore.seed
    mindspore.set_rng_state

序列化
-------

.. mscnautosummary::
    :toctree: mindspore

    mindspore.async_ckpt_thread_status
    mindspore.build_searched_strategy
    mindspore.check_checkpoint
    mindspore.convert_model
    mindspore.export
    mindspore.load
    mindspore.load_checkpoint
    mindspore.load_checkpoint_async
    mindspore.load_distributed_checkpoint
    mindspore.load_mindir
    mindspore.load_param_into_net
    mindspore.load_segmented_checkpoints
    mindspore.merge_pipeline_strategys
    mindspore.merge_sliced_parameter
    mindspore.obfuscate_model
    mindspore.parse_print
    mindspore.rank_list_for_transform
    mindspore.restore_group_info_list
    mindspore.save_checkpoint
    mindspore.save_mindir
    mindspore.transform_checkpoint_by_rank
    mindspore.transform_checkpoints

自动微分
----------------

.. mscnautosummary::
    :toctree: mindspore

    mindspore.grad
    mindspore.value_and_grad
    mindspore.get_grad
    mindspore.jacfwd
    mindspore.jacrev
    mindspore.jvp
    mindspore.vjp

并行优化
---------

自动向量化
^^^^^^^^^^^

.. mscnautosummary::
    :toctree: mindspore

    mindspore.vmap

并行
^^^^^

.. mscnautosummary::
    :toctree: mindspore

    mindspore.Layout
    mindspore.parameter_broadcast
    mindspore.recompute
    mindspore.reshard
    mindspore.shard
    mindspore.sync_pipeline_shared_parameters

即时编译
--------

.. mscnautosummary::
    :toctree: mindspore

    mindspore.JitConfig
    mindspore.jit
    mindspore.jit_class
    mindspore.ms_class
    mindspore.ms_function
    mindspore.ms_memory_recycle
    mindspore.mutable
    mindspore.constexpr
    mindspore.lazy_inline
    mindspore.no_inline

工具
-----

数据处理工具
^^^^^^^^^^^^^

.. mscnautosummary::
    :toctree: mindspore

    mindspore.DatasetHelper
    mindspore.Symbol
    mindspore.connect_network_with_dataset
    mindspore.data_sink

调试调优
^^^^^^^^^

.. mscnautosummary::
    :toctree: mindspore

    mindspore.Profiler
    mindspore.SummaryCollector
    mindspore.SummaryLandscape
    mindspore.SummaryRecord
    mindspore.set_dump

日志
^^^^^

.. mscnautosummary::
    :toctree: mindspore

    mindspore.get_level
    mindspore.get_log_config


安装验证
^^^^^^^^^

.. mscnautosummary::
    :toctree: mindspore

    mindspore.run_check