tf.experimental.dlpack.to_dlpack

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

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.