Save the date! Google I/O returns May 18-20 Register now

tf.experimental.dlpack.from_dlpack

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