原文： PyTorch torch.utils.dlpack

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

將 DLPack 解碼為張量。

參數(shù)

dlpack –具有 dltensor 的 PyCapsule 對(duì)象

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

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

返回表示張量的 DLPack。

Parameters

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

dlpack 共享張量?jī)?nèi)存。 請(qǐng)注意，每個(gè) dlpack 只能使用一次。