jax.dlpack module

Warning

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

jax.dlpack module¶

from_dlpack(dlpack)

Returns a DeviceArray representation of a DLPack tensor dlpack.

to_dlpack(x[, take_ownership])

Returns a DLPack tensor that encapsulates a DeviceArray x.

previous

jax.numpy_rank_promotion

next

jax.dlpack.from_dlpack