tf.experimental.dtensor.create_mesh

Создает сетку для одного клиента.

tf.experimental.dtensor.create_mesh(
    mesh_dims: Optional[Union[List[Tuple[str, int]], Dict[str, int]]] = None,
    mesh_name: str = '',
    devices: Optional[List[Union[tf_device.DeviceSpec, str]]] = None,
    device_type: Optional[str] = None,
    use_xla_spmd: bool = layout.USE_XLA_SPMD
) -> tf.experimental.dtensor.Mesh

Используется в блокнотах

Используется в руководстве	Используется в учебных пособиях
Основные понятия DTensor Распределенное обучение с API Core и DTensor	Распределенное обучение с DTensor Использование DTensor с Keras

Если указаны и mesh_dims, и devices, они должны совпадать. В особом случае, когда все аргументы отсутствуют, создается одномерная сетка CPU с пустым именем, и все доступные устройства назначаются этому измерению.

Аргументы
`mesh_dims`	Словарь dim_name: dim_size или список кортежей (dim_name, dim_size). По умолчанию используется одно измерение пакетного параллелизма с именем 'x', задействующее все устройства. В особом случае сетка mesh_dims из одного элемента с dim_size, равным -1, также использует все устройства. Например, `{'x' : 4, 'y' : 1}` или `[('x', 4), ('y', 1)]`.
`mesh_name`	Имя созданной сетки. По умолчанию — ''.
`devices`	Строковые представления используемых устройств. Это часть устройства в tf.DeviceSpec, например 'CPU:0'. По умолчанию используются все доступные логические устройства.
`device_type`	Если `devices` отсутствует, указывает тип используемых устройств. По умолчанию — 'CPU'.
`use_xla_spmd`	Если значение равно True, будет использоваться XLA SPMD вместо DTensor SPMD.

Возвращает
Сетка для одного клиента, созданная с указанными аргументами или аргументами по умолчанию.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/create_mesh