|View source on GitHub|
Returns the dlpack capsule representing the tensor.
tf.experimental.dlpack.to_dlpack( tf_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
tf_tensor: Tensorflow eager tensor, to be converted to dlpack capsule.
A PyCapsule named as dltensor, which shares the underlying memory to other framework. This PyCapsule can be consumed only once.