mindspore.Layout
- class mindspore.Layout(device_matrix, alias_name)[源代码]
Layout描述了详细的切分信息。
说明
仅在半自动并行或自动并行模式下有效。
device_matrix 的累乘结果必须等于一个pipeline stage中的设备数。
当Layout来构建切分策略时,每个别名只允许用于一次张量的切分。
- 参数:
device_matrix (tuple) - 描述设备排列的形状,其元素类型为int。
alias_name (tuple) - device_matrix的每个轴的别名,其元素类型为字符串。
- 异常:
TypeError - device_matrix 不是元组类型。
TypeError - alias_name 不是元组类型。
ValueError - device_matrix 长度不等于 alias_name 长度。
TypeError - device_matrix 的元素不是int类型。
TypeError - alias_name 的元素不是str类型。
ValueError - alias_name 的元素是一个空的str。
ValueError - alias_name 的元素为“None”。
ValueError - alias_name 包含重复的元素。
样例:
>>> from mindspore import Layout >>> layout = Layout((2, 2, 2), ("dp", "sp", "mp")) >>> layout0 = layout("dp", "mp") >>> print(layout0.to_dict()) {"device_matrix": (2, 2, 2), "tensor_map": (2, 0)}