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 и DTensor	Распределённое обучение с DTensors Использование DTensors с Keras

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

Аргументы
`mesh_dims`	Словарь dim_name: dim_size или список кортежей (dim_name, dim_size). По умолчанию используется единственная размерность для пакетного параллелизма, называемая 'x', используя все устройства. В качестве частного случая, меш_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