tf.experimental.dtensor.create_mesh
Создаёт меш для одного клиента.
tf.experimental.dtensor.create_mesh(
mesh_dims: Optional[List[Tuple[str, int]]] = None,
mesh_name: str = '',
devices: Optional[List[str]] = None,
device_type: Optional[str] = None
) -> tf.experimental.dtensor.Mesh
Если заданы и mesh_dims , и devices, они должны совпадать. В качестве специального случая, при отсутствии всех аргументов, создаётся 1D меш CPU с пустым именем, которому присваиваются все доступные устройства для этого измерения.
| Аргументы | |
|---|---|
mesh_dims | Список кортежей (dim_name, dim_size). По умолчанию используется одно измерение для параллельной обработки по партиям с именем 'x', используя все устройства. В качестве специального случая, если у mesh_dims есть один элемент с dim_size равным -1, используются все устройства. |
mesh_name | Имя создаваемого меша. По умолчанию ''. |
devices | Строковое представление устройств для использования. Это часть tf.DeviceSpec, например, 'CPU:0'. По умолчанию используются все доступные логические устройства. |
device_type | Если devices отсутствует, тип используемых устройств. По умолчанию 'CPU'. |
| Возвращаемое значение | |
|---|---|
| Меш для одного клиента, созданный из указанных или стандартных аргументов. |
© 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/versions/r2.9/api_docs/python/tf/experimental/dtensor/create_mesh
