tf.experimental.dlpack.from_dlpack

View source on GitHub

Returns the Tensorflow eager tensor.

tf.experimental.dlpack.from_dlpack(
    dlcapsule
)

The returned tensor uses the memory shared by dlpack capsules from other framework.

a = tf.experimental.dlpack.from_dlpack(dlcapsule)
# `a` uses the memory shared by dlpack

Args:

  • dlcapsule: A PyCapsule named as dltensor

Returns:

A Tensorflow eager tensor