tf.experimental.dtensor.is_dtensor
Проверка, является ли входной тензор DTensor.
tf.experimental.dtensor.is_dtensor(
tensor
) -> bool
В Python DTensor имеет тот же тип, что и tf.Tensor. Этот метод позволит вам проверить и обработать тензор по-разному, если tf.Tensor является DTensor.
| Аргументы | |
|---|---|
tensor | объект, который нужно проверить. |
| Возвращаемое значение | |
|---|---|
| bool, True, если данный тензор является 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/is_dtensor
