tf.experimental.dlpack.from_dlpack

View source on GitHub

Returns the Tensorflow eager tensor.

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

dlcapsule A PyCapsule named as dltensor

A Tensorflow eager tensor