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 Распределённое обучение с помощью основных API и DTensor	Распределённое обучение с DTensors

Если layout не равно None, fn потребляет 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