jax.dlpack.from_dlpack

jax.dlpack.from_dlpack(dlpack)

Returns a DeviceArray representation of a DLPack tensor.

The returned DeviceArray shares memory with dlpack.

Parameters

dlpack – a DLPack tensor, on either CPU or GPU.