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")