PyTorch torch.utils.dlpack

原文： PyTorch torch.utils.dlpack

torch.utils.dlpack.from_dlpack(dlpack) → Tensor?

將 DLPack 解碼為張量。

參數(shù)

dlpack –具有 dltensor 的 PyCapsule 對象

張量將與 dlpack 中表示的對象共享內(nèi)存。 請注意，每個 dlpack 只能使用一次。

torch.utils.dlpack.to_dlpack(tensor) → PyCapsule?

返回表示張量的 DLPack。

Parameters

張量 –要導(dǎo)出的張量

dlpack 共享張量內(nèi)存。 請注意，每個 dlpack 只能使用一次。