dlpack.rst

DLPack

Via a DLPack capsule, you can borrow a tensor from external software as a :obj:`nnabla.NdArray` and can share a NdArray to external software.

.. automodule:: nnabla.utils.dlpack

.. autofunction:: from_dlpack

.. autofunction:: to_dlpack