TUTORIALAI 与数据科学FETCHED
PyTorch torch.from_dlpack 函数
PyTorch torch.from_dlpack 函数 Pytorch torch 参考手册 torch.from_dlpack 是 PyTorch 中用于从 dlpack 张量创建 PyTorch 张量的函数。 函数定义 torch.from_dlpack(ext_tensor) 使用示例 实例 [mycode4 type='python'] import torch # 创建张量 x = ..
更新于 2026/7/16
教程详情页默认 noindex。
PyTorch torch.from_dlpack 函数
Pytorch torch 参考手册torch.from_dlpack 是 PyTorch 中用于从 dlpack 张量创建 PyTorch 张量的函数。
函数定义
torch.from_dlpack(ext_tensor)
使用示例
实例
import torch
# 创建张量
x = torch.randn(3, 4)
# 获取 dlpack 胶囊
dlpack = x.toDlpack()
# 从 dlpack 创建新张量
y = torch.from_dlpack(dlpack)
print(y.shape)
# 创建张量
x = torch.randn(3, 4)
# 获取 dlpack 胶囊
dlpack = x.toDlpack()
# 从 dlpack 创建新张量
y = torch.from_dlpack(dlpack)
print(y.shape)