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

Если 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/versions/r2.9/api_docs/python/tf/experimental/dtensor/call_with_layout