tf.experimental.dlpack.to_dlpack

Возвращает dlpack капсулу, представляющую тензор.

tf.experimental.dlpack.to_dlpack(
    tf_tensor
)

Эта операция гарантирует, что основная память данных готова к возврату.

a = tf.tensor([1, 10])
dlcapsule = tf.experimental.dlpack.to_dlpack(a)
# dlcapsule represents the dlpack data structure

Аргументы
`tf_tensor`	Тензор TensorFlow eager, который нужно преобразовать в dlpack капсулу.

Возвращаемое значение
PyCapsule с именем dltensor, который разделяет основную память с другими фреймворками. Эту PyCapsule можно использовать только один раз.

© 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/dlpack/to_dlpack