tf.experimental.dtensor.call_with_layout

Вызывает функцию в области действия устройства DTensor, если layout не равен None.

tf.experimental.dtensor.call_with_layout(
    fn: Callable[..., Any],
    layout: Optional[tf.experimental.dtensor.Layout],
    *args,
    **kwargs
) -> Any

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

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

Если layout не равен None, fn принимает DTensor (или несколько DTensor) в качестве входных данных и возвращает DTensor; DTensor — это tf.Tensor с атрибутами, связанными с компоновкой.

Если layout равен None, fn принимает и возвращает обычные tf.Tensor.

Аргументы
`fn`	Поддерживаемая функция TF API, например tf.zeros.
`layout`	Необязательная компоновка выходного DTensor.
`*args`	Аргументы, передаваемые в `fn`.
`**kwargs`	Именованные аргументы, передаваемые в `fn`.

Возвращает
Возвращаемое значение `fn`, преобразованное в DTensor, если это запрошено.

© 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/call_with_layout