Модуль: tf.experimental.dlpack
Общедоступный API для пространства имён tf._api.v2.experimental.dlpack
Функции
from_dlpack(...): Возвращает тензор eager TensorFlow.
to_dlpack(...): Возвращает капсулу dlpack, представляющую тензор.
© 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/dlpack