jax.dlpack.from_dlpack

Warning

This page was created from a pull request (#9655).

jax.dlpack.from_dlpack¶

jax.dlpack.from_dlpack(dlpack)[source]¶

Returns a DeviceArray representation of a DLPack tensor dlpack.

The returned DeviceArray shares memory with dlpack.

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

previous

jax.dlpack module

next

jax.dlpack.to_dlpack