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, который разделяет основную память с другими фреймворками. Эту капсулу можно использовать только один раз.

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