tf.experimental.dlpack.to_dlpack

Returns the dlpack capsule representing the tensor.

This operation ensures the underlying data memory is ready when returns.

a = tf.tensor([1, 10])
dlcapsule = tf.experimental.dlpack.to_dlpack(a)
# dlcapsule represents the dlpack data structure
Args
tf_tensor Tensorflow eager tensor, to be converted to dlpack capsule.
Returns
A PyCapsule named as dltensor, which shares the underlying memory to other framework. This PyCapsule can be consumed only once.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/experimental/dlpack/to_dlpack