mindspore.get_auto_parallel_context

mindspore.get_auto_parallel_context(attr_key)[源代码]

根据key获取自动并行的配置。

参数:

  • attr_key (str) - 配置的key。

返回:

根据key返回配置的值。

异常:

ValueError:输入key不在自动并行的配置列表中。

样例:

>>> import mindspore as ms
>>> parallel_mode = ms.get_auto_parallel_context("parallel_mode")
>>> dataset_strategy = ms.get_auto_parallel_context("dataset_strategy")