codecamp

PyTorch torch.utils.dlpack

原文： PyTorch torch.utils.dlpack

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

将 DLPack 解码为张量。

参数

dlpack –具有 dltensor 的 PyCapsule 对象

张量将与 dlpack 中表示的对象共享内存。 请注意，每个 dlpack 只能使用一次。

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

返回表示张量的 DLPack。

Parameters

张量 –要导出的张量

dlpack 共享张量内存。 请注意，每个 dlpack 只能使用一次。

PyTorch torch.utils.data
PyTorch torch.utils.model_zoo
温馨提示
下载编程狮App，免费阅读超1000+编程语言教程
取消
确定
目录

PyTorch 入门

PyTorch深度学习

PyTorch 图片

Pytorch 音频

Pytorch 文本

PyTorch 命名为 Tensor(实验性）

PyTorch 强化学习

PyTorch 在生产中部署 PyTorch 模型

PyTorch 并行和分布式训练

PyTorch 扩展

PyTorch 模型优化

PyTorch 用其他语言

PyTorch 基础知识

PyTorch 笔记

PyTorch 语言绑定

Python API

PyTorch torchvision参考

PyTorch 音频参考

PyTorch 社区

关闭

 MIP.setData({ 'pageTheme' : getCookie('pageTheme') || {'day':true, 'night':false}, 'pageFontSize' : getCookie('pageFontSize') || 20 }); MIP.watch('pageTheme', function(newValue){ setCookie('pageTheme', JSON.stringify(newValue)) }); MIP.watch('pageFontSize', function(newValue){ setCookie('pageFontSize', newValue) }); function setCookie(name, value){ var days = 1; var exp = new Date(); exp.setTime(exp.getTime() + days*24*60*60*1000); document.cookie = name + '=' + value + ';expires=' + exp.toUTCString(); } function getCookie(name){ var reg = new RegExp('(^| )' + name + '=([^;]*)(;|$)'); return document.cookie.match(reg) ? JSON.parse(document.cookie.match(reg)[2]) : null; }